Rthur2003 commited on
Commit
026a187
·
1 Parent(s): 4324aed

feat: add TorchSklearnWrapper import for model deserialization support

Browse files
Files changed (1) hide show
  1. local_demo.py +1 -0
local_demo.py CHANGED
@@ -48,6 +48,7 @@ except ModuleNotFoundError as exc:
48
  import numpy as np
49
 
50
  from app.training.extract_features_batch import extract_sample_features
 
51
 
52
  BASE_DIR = Path(__file__).resolve().parent
53
  PROJECT_ROOT = BASE_DIR.parent
 
48
  import numpy as np
49
 
50
  from app.training.extract_features_batch import extract_sample_features
51
+ from app.training.train_deep_classifiers import TorchSklearnWrapper # noqa: F401 — required for pickle deserialization of DL models
52
 
53
  BASE_DIR = Path(__file__).resolve().parent
54
  PROJECT_ROOT = BASE_DIR.parent