Skip to content

Commit d2d8311

Browse files
Merge pull request #628 from guillaume-vignal/fix_report
Fix `display_model_analysis` in Shapash Report to correctly retrieve `sklearn` version
2 parents 767ca75 + eb08c0c commit d2d8311

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

shapash/report/project_report.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,13 +235,20 @@ def display_model_analysis(self):
235235
print_md(f"**Library :** {self.explainer.model.__class__.__module__}")
236236

237237
for _, module in sorted(sys.modules.items()):
238+
if not hasattr(module, "__name__"):
239+
continue
240+
238241
module_name = module.__name__.split(".")[0]
239-
if self.explainer.model.__class__.__module__.split(".")[0] == module_name:
242+
expected_name = self.explainer.model.__class__.__module__.split(".")[0]
243+
244+
if expected_name == module_name:
240245
try:
241-
version = importlib.metadata.version(module_name)
246+
package_name = "scikit-learn" if module_name == "sklearn" else module_name
247+
version = importlib.metadata.version(package_name)
242248
print_md(f"**Library version :** {version}")
243249
except importlib.metadata.PackageNotFoundError:
244250
print_md(f"**Library version :** not found for {module_name}")
251+
break
245252

246253
print_md("**Model parameters :** ")
247254
model_params = self.explainer.model.__dict__

0 commit comments

Comments
 (0)