updated classification.py

pycaret/classification.py

@@ -706,8 +706,8 @@ def setup(data,
     from IPython.display import display, HTML, clear_output, update_display
     
     #import mlflow and logging utils
-    import mlflow
-    import mlflow.sklearn
+    #import mlflow
+    #import mlflow.sklearn
     import secrets
     import os
     
@@ -1651,6 +1651,8 @@ def setup(data,
 
     if logging_param:
 
+        import mlflow
+
         if experiment_name is None:
             exp_name_ = 'clf-default-name'
         else:
@@ -3965,6 +3967,9 @@ def compare_models(blacklist = None,
     else:
         n_select_num = abs(n_select)
 
+    if n_select_num > len_mod:
+        n_select_num = len_mod
+
     if whitelist is not None:
         wl = len(whitelist)
         bl = len_of_blacklist