Browse Source

Merge pull request #105 from Learnware-LAMDA/fix_bug_bxdd2

[FIX] fix package utils bug
tags/v0.3.2
bxdd GitHub 2 years ago
parent
commit
d94c4d0d2b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 2 deletions
  1. +0
    -1
      learnware/client/container.py
  2. +1
    -1
      learnware/client/package_utils.py

+ 0
- 1
learnware/client/container.py View File

@@ -123,7 +123,6 @@ class ModelCondaContainer(ModelContainer):

input_shape = output_results["metadata"]["input_shape"]
output_shape = output_results["metadata"]["output_shape"]
logger.info(f"input_shape: {input_shape}, output_shape: {output_shape}")
self.reset(input_shape=input_shape, output_shape=output_shape)

def _run_model_with_script(self, method, **kargs):


+ 1
- 1
learnware/client/package_utils.py View File

@@ -78,7 +78,7 @@ def filter_nonexist_pip_packages(packages: list) -> Tuple[List[str], List[str]]:
continue
try:
package_name = parse_pip_requirement(package)
if package_name != "learnware":
if package_name is not None and package_name != "learnware":
try_to_run(args=["pip", "index", "versions", package_name], timeout=5)
exist_packages.append(package)
continue


Loading…
Cancel
Save