Преглед изворни кода

Fix PyTorch warning supression (#502)

Co-authored-by: Alex <alexandershulga.sh@gmail.com>
(cherry picked from commit bb3aed6cc69bf5aad832a1b59db41b8d627e69e7)
Alexander Borzunov пре 3 година
родитељ
комит
df1508aad3
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      hivemind/compression/base.py

+ 1 - 1
hivemind/compression/base.py

@@ -11,7 +11,7 @@ from hivemind.proto import runtime_pb2
 from hivemind.utils.tensor_descr import TensorDescriptor
 
 # While converting read-only NumPy arrays into PyTorch tensors, we don't make extra copies for efficiency
-warnings.filterwarnings("ignore", message="The given NumPy array is not writeable", category=UserWarning)
+warnings.filterwarnings("ignore", message="The given NumPy array is not writable", category=UserWarning)
 
 
 Key = Any