yingda.chen 3 years ago
parent
commit
78ef56cba3
3 changed files with 6 additions and 1 deletions
  1. +3
    -0
      modelscope/models/science/unifold/modules/__init__.py
  2. +2
    -0
      requirements/science.txt
  3. +1
    -1
      tests/pipelines/test_unifold.py

+ 3
- 0
modelscope/models/science/unifold/modules/__init__.py View File

@@ -0,0 +1,3 @@
# The Uni-fold implementation is also open-sourced by the authors under Apache-2.0 license,
# and is publicly available at https://github.com/dptech-corp/Uni-Fold.
"""Unifold Modules."""

+ 2
- 0
requirements/science.txt View File

@@ -1,4 +1,6 @@
biopython
iopath
ipdb
lmdb
ml_collections
scipy


+ 1
- 1
tests/pipelines/test_unifold.py View File

@@ -19,7 +19,7 @@ class UnifoldProteinStructureTest(unittest.TestCase, DemoCompatibilityCheck):
self.protein_multimer = 'GAMGLPEEPSSPQESTLKALSLYEAHLSSYIMYLQTFLVKTKQKVNNKNYPEFTLFDTSKLKKDQTLKSIKT' + \
'NIAALKNHIDKIKPIAMQIYKKYSKNIP'

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_by_direct_model_download(self):
model_dir = snapshot_download(self.model_id)
mono_pipeline_ins = pipeline(task=self.task, model=model_dir)


Loading…
Cancel
Save