Browse Source

[FIX] fix test workflow

tags/v0.3.2
bxdd 2 years ago
parent
commit
77f8dffa15
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      tests/test_workflow/test_workflow.py

+ 3
- 1
tests/test_workflow/test_workflow.py View File

@@ -18,7 +18,7 @@ import learnware.specification as specification
curr_root = os.path.dirname(os.path.abspath(__file__))

user_semantic = {
"Data": {"Values": ["Image"], "Type": "Class"},
"Data": {"Values": ["Table"], "Type": "Class"},
"Task": {
"Values": ["Classification"],
"Type": "Class",
@@ -96,6 +96,8 @@ class TestAllWorkflow(unittest.TestCase):
semantic_spec = copy.deepcopy(user_semantic)
semantic_spec["Name"]["Values"] = "learnware_%d" % (idx)
semantic_spec["Description"]["Values"] = "test_learnware_number_%d" % (idx)
semantic_spec["Input"] = {"Dimension": 64}
semantic_spec["Input"].update({f"{i}": f"The value in the digit image with row is {i // 8} and col is {i % 8}." for i in range(64)})
semantic_spec["Output"] = {"Dimension": 1, "Description": {"0": "The label of the hand-written digit."}}
easy_market.add_learnware(zip_path, semantic_spec)



Loading…
Cancel
Save