Skip to content

Commit 75ecc34

Browse files
committed
undo dtype changes.
1 parent 9c4bd06 commit 75ecc34

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/ptwt/constants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
"WaveletTensorTuple",
2424
]
2525

26-
SUPPORTED_DTYPES = {torch.float32, torch.float64, torch.float16, torch.bfloat16}
26+
SUPPORTED_DTYPES = {torch.float32, torch.float64}
2727

2828

2929
class Wavelet(Protocol):

0 commit comments

Comments
 (0)