File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff 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__
You can’t perform that action at this time.
0 commit comments