Browse Source

[FIX] fix import bugs, and update setup

tags/v0.3.2
bxdd 2 years ago
parent
commit
ac2c4bb4c5
2 changed files with 4 additions and 3 deletions
  1. +1
    -1
      learnware/specification/system/__init__.py
  2. +3
    -2
      setup.py

+ 1
- 1
learnware/specification/system/__init__.py View File

@@ -8,4 +8,4 @@ if not is_torch_available(verbose=False):
HeteroMapTableSpecification = None
logger.warning("HeteroMapTableSpecification is skipped because torch is not installed!")
else:
from .rkme import HeteroMapTableSpecification
from .hetero_table import HeteroMapTableSpecification

+ 3
- 2
setup.py View File

@@ -107,8 +107,9 @@ if __name__ == "__main__":
"pre-commit",
],
"full": [
"torch>=2.0.0",
"torchvision>=0.15.1",
# The default full requirements for learnware package
"torch==2.1.0",
"torchvision==0.16.0",
"torch-optimizer>=0.3.0",
"lightgbm>=3.3.0",
"sentence_transformers>=2.2.2",


Loading…
Cancel
Save