Browse Source

[MNT] solve E722,F402 error of flake8 in tests

tags/v0.3.2
Gene 2 years ago
parent
commit
08ba9320d3
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      tests/test_learnware_client/test_all_learnware.py
  2. +2
    -2
      tests/test_workflow/test_hetero_workflow.py

+ 1
- 1
tests/test_learnware_client/test_all_learnware.py View File

@@ -56,7 +56,7 @@ class TestAllLearnware(unittest.TestCase):
semantic_spec = self.client.get_semantic_specification(idx)
LearnwareClient.check_learnware(zip_path, semantic_spec)
print(f"check learnware {idx} succeed")
except:
except Exception:
failed_ids.append(idx)
print(f"check learnware {idx} failed!!!")



+ 2
- 2
tests/test_workflow/test_hetero_workflow.py View File

@@ -295,8 +295,8 @@ class TestHeteroWorkflow(unittest.TestCase):

# multi model reuse
hetero_learnware_list = []
for learnware in multiple_result[0].learnwares:
hetero_learnware = HeteroMapAlignLearnware(learnware, mode="regression")
for org_learnware in multiple_result[0].learnwares:
hetero_learnware = HeteroMapAlignLearnware(org_learnware, mode="regression")
hetero_learnware.align(user_spec, X[:100], y[:100])
hetero_learnware_list.append(hetero_learnware)



Loading…
Cancel
Save