VibecoderMcSwaggins commited on
Commit
0cf2de1
·
1 Parent(s): e33b9b5

fix: Import load_model_from_npz from serialization (no Hydra dependency)

Browse files

Fixes runtime error: ModuleNotFoundError: No module named 'hydra'

Changed internal imports to avoid pulling in Hydra for inference-only code.
load_model_from_npz lives in serialization.py which has no Hydra dependency.

src/antibody_training_esm/core/__init__.py CHANGED
@@ -10,7 +10,7 @@ Professional ML components for antibody classification:
10
 
11
  from antibody_training_esm.core.classifier import BinaryClassifier
12
  from antibody_training_esm.core.embeddings import ESMEmbeddingExtractor
13
- from antibody_training_esm.core.trainer import load_model_from_npz
14
 
15
  __all__ = [
16
  "BinaryClassifier",
 
10
 
11
  from antibody_training_esm.core.classifier import BinaryClassifier
12
  from antibody_training_esm.core.embeddings import ESMEmbeddingExtractor
13
+ from antibody_training_esm.core.training.serialization import load_model_from_npz
14
 
15
  __all__ = [
16
  "BinaryClassifier",
src/antibody_training_esm/core/__pycache__/__init__.cpython-312.pyc CHANGED
Binary files a/src/antibody_training_esm/core/__pycache__/__init__.cpython-312.pyc and b/src/antibody_training_esm/core/__pycache__/__init__.cpython-312.pyc differ
 
src/antibody_training_esm/core/__pycache__/prediction.cpython-312.pyc CHANGED
Binary files a/src/antibody_training_esm/core/__pycache__/prediction.cpython-312.pyc and b/src/antibody_training_esm/core/__pycache__/prediction.cpython-312.pyc differ
 
src/antibody_training_esm/core/prediction.py CHANGED
@@ -12,7 +12,7 @@ from sklearn.linear_model import LogisticRegression
12
  from antibody_training_esm.core.classifier import BinaryClassifier
13
  from antibody_training_esm.core.config import DEFAULT_BATCH_SIZE
14
  from antibody_training_esm.core.embeddings import ESMEmbeddingExtractor
15
- from antibody_training_esm.core.trainer import load_model_from_npz
16
  from antibody_training_esm.models.prediction import (
17
  AssayType,
18
  PredictionRequest,
 
12
  from antibody_training_esm.core.classifier import BinaryClassifier
13
  from antibody_training_esm.core.config import DEFAULT_BATCH_SIZE
14
  from antibody_training_esm.core.embeddings import ESMEmbeddingExtractor
15
+ from antibody_training_esm.core.training.serialization import load_model_from_npz
16
  from antibody_training_esm.models.prediction import (
17
  AssayType,
18
  PredictionRequest,