Skip to content

Commit 9651415

Browse files
committed
Remove __main_model__ interface in estimators
1 parent 90c8d4d commit 9651415

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

gluefactory/robust_estimators/__init__.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,7 @@ def load_estimator(type, estimator):
1919
try:
2020
return get_class(path, BaseEstimator)
2121
except AssertionError:
22-
mod = __import__(path, fromlist=[""])
23-
try:
24-
return mod.__main_model__
25-
except AttributeError as exc:
26-
print(exc)
27-
continue
22+
continue
2823
raise RuntimeError(
2924
f'Model {estimator} not found in any of [{" ".join(import_paths)}]'
3025
)

0 commit comments

Comments
 (0)