Browse Source

[MNT] modify import

tags/v0.3.2
bxdd 2 years ago
parent
commit
0edbeabe07
8 changed files with 12 additions and 7 deletions
  1. +2
    -1
      examples/dataset_image_workflow/main.py
  2. +2
    -1
      examples/dataset_pfs_workflow/main.py
  3. +1
    -1
      examples/workflow_by_code/main.py
  4. +3
    -0
      learnware/reuse/__init__.py
  5. +1
    -1
      tests/test_learnware_client/test_docker.py
  6. +1
    -1
      tests/test_learnware_client/test_load.py
  7. +1
    -1
      tests/test_learnware_client/test_reuse.py
  8. +1
    -1
      tests/test_workflow/test_workflow.py

+ 2
- 1
examples/dataset_image_workflow/main.py View File

@@ -4,7 +4,8 @@ from get_data import *
import os
import random
from utils import generate_uploader, generate_user, ImageDataLoader, train, eval_prediction
from learnware.learnware import Learnware, JobSelectorReuser, AveragingReuser
from learnware.learnware import Learnware
from learnware.reuse import JobSelectorReuser, AveragingReuser
import time

from learnware.market import EasyMarket, BaseUserInfo


+ 2
- 1
examples/dataset_pfs_workflow/main.py View File

@@ -9,7 +9,8 @@ from shutil import copyfile, rmtree
import learnware
from learnware.market import EasyMarket, BaseUserInfo
from learnware.market import database_ops
from learnware.learnware import Learnware, JobSelectorReuser, AveragingReuser
from learnware.learnware import Learnware
from learnware.reuse import JobSelectorReuser, AveragingReuser
import learnware.specification as specification
from pfs import Dataloader
from learnware.logger import get_module_logger


+ 1
- 1
examples/workflow_by_code/main.py View File

@@ -11,7 +11,7 @@ from shutil import copyfile, rmtree

import learnware
from learnware.market import EasyMarket, BaseUserInfo
from learnware.learnware import JobSelectorReuser, AveragingReuser
from learnware.reuse import JobSelectorReuser, AveragingReuser
import learnware.specification as specification
from learnware.utils import get_module_by_module_path



+ 3
- 0
learnware/reuse/__init__.py View File

@@ -0,0 +1,3 @@
from .ensemble_pruning import EnsemblePruningReuser
from .averaging import AveragingReuser
from .job_selector import JobSelectorReuser

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

@@ -5,7 +5,7 @@ import numpy as np
import learnware
from learnware.client import LearnwareClient
from learnware.client.container import LearnwaresContainer
from learnware.learnware.reuse import AveragingReuser
from learnware.reuse import AveragingReuser


if __name__ == "__main__":


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

@@ -7,7 +7,7 @@ import learnware
from learnware.learnware import get_learnware_from_dirpath
from learnware.client import LearnwareClient
from learnware.client.container import ModelCondaContainer, LearnwaresContainer
from learnware.learnware.reuse import AveragingReuser
from learnware.reuse import AveragingReuser


class TestLearnwareLoad(unittest.TestCase):


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

@@ -3,7 +3,7 @@ import numpy as np

from learnware.learnware import get_learnware_from_dirpath
from learnware.client.container import LearnwaresContainer
from learnware.learnware.reuse import AveragingReuser
from learnware.reuse import AveragingReuser
from learnware.test.module import get_semantic_specification

if __name__ == "__main__":


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

@@ -12,7 +12,7 @@ from shutil import copyfile, rmtree

import learnware
from learnware.market import EasyMarket, BaseUserInfo
from learnware.learnware import JobSelectorReuser, AveragingReuser, EnsemblePruningReuser
from learnware.reuse import JobSelectorReuser, AveragingReuser, EnsemblePruningReuser
import learnware.specification as specification

curr_root = os.path.dirname(os.path.abspath(__file__))


Loading…
Cancel
Save