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
- # precollator functions
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
  """