Christina Theodoris
commited on
Commit
·
9d8d2f0
1
Parent(s):
aea4988
silence tensor copy warning
Browse files
geneformer/collator_for_classification.py
CHANGED
@@ -26,8 +26,9 @@ LARGE_INTEGER = int(
|
|
26 |
1e20
|
27 |
) # This is used when we need something big but slightly smaller than VERY_LARGE_INTEGER
|
28 |
|
29 |
-
|
30 |
|
|
|
31 |
|
32 |
class ExplicitEnum(Enum):
|
33 |
"""
|
|
|
26 |
1e20
|
27 |
) # This is used when we need something big but slightly smaller than VERY_LARGE_INTEGER
|
28 |
|
29 |
+
warnings.filterwarnings("ignore", message="To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach()", category=UserWarning, module="torch")
|
30 |
|
31 |
+
# precollator functions
|
32 |
|
33 |
class ExplicitEnum(Enum):
|
34 |
"""
|