From 2a9ec6d1e7347f33a5c0fdb535ed4c479f43489f Mon Sep 17 00:00:00 2001 From: Ben Auffarth Date: Tue, 17 Dec 2019 14:12:51 +0000 Subject: [PATCH] version number to 1.0 if unknown hack: set version number to 1.0 if not version number available. Missing version numbers lead to failing imports of sklearn models --- openml/extensions/sklearn/extension.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/openml/extensions/sklearn/extension.py b/openml/extensions/sklearn/extension.py index 9720bd853..da73a0f73 100644 --- a/openml/extensions/sklearn/extension.py +++ b/openml/extensions/sklearn/extension.py @@ -763,7 +763,12 @@ def _get_external_version_string( # sorted concatenation of all modules which are present in this run. model_package_name = model.__module__.split('.')[0] module = importlib.import_module(model_package_name) - model_package_version_number = module.__version__ # type: ignore + if '__version__' in module.__dict__: + model_package_version_number = module.__version__ # type: ignore + else: + print('{} does not have a version number!'.format(module)) + print('setting to 1.0. This is a hack.') + model_package_version_number = 1.0 external_version = self._format_external_version( model_package_name, model_package_version_number, )