makiling commited on
Commit
7420f23
·
verified ·
1 Parent(s): aab1198

Upload folder using huggingface_hub

Browse files
polyreact/features/__pycache__/plm.cpython-311.pyc CHANGED
Binary files a/polyreact/features/__pycache__/plm.cpython-311.pyc and b/polyreact/features/__pycache__/plm.cpython-311.pyc differ
 
polyreact/features/plm.py CHANGED
@@ -3,6 +3,7 @@
3
  from __future__ import annotations
4
 
5
  import hashlib
 
6
  from concurrent.futures import ThreadPoolExecutor
7
  from dataclasses import dataclass
8
  from pathlib import Path
@@ -24,6 +25,13 @@ try: # pragma: no cover - optional dependency
24
  except ImportError: # pragma: no cover - optional dependency
25
  esm = None
26
 
 
 
 
 
 
 
 
27
  from .anarsi import AnarciNumberer
28
 
29
  ModelLoader = Callable[[str, str], Tuple[object, nn.Module]]
 
3
  from __future__ import annotations
4
 
5
  import hashlib
6
+ import warnings
7
  from concurrent.futures import ThreadPoolExecutor
8
  from dataclasses import dataclass
9
  from pathlib import Path
 
25
  except ImportError: # pragma: no cover - optional dependency
26
  esm = None
27
 
28
+ if esm is not None: # pragma: no cover - optional dependency
29
+ warnings.filterwarnings(
30
+ "ignore",
31
+ message="Regression weights not found, predicting contacts will not produce correct results.",
32
+ module="esm.pretrained",
33
+ )
34
+
35
  from .anarsi import AnarciNumberer
36
 
37
  ModelLoader = Callable[[str, str], Tuple[object, nn.Module]]