Merge pull request pycaret#1128 from markgrujic/patch-1

Fix load_model() for AWS models in persistence.py
s0ldier03 · Mar 24, 2021 · 37f458b · 37f458b
2 parents f375479 + b5f038a
commit 37f458b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycaret/internal/persistence.py b/pycaret/internal/persistence.py
@@ -393,7 +393,7 @@ def load_model(
                 os.makedirs(path)
             s3.Bucket(bucketname).download_file(key, filename)
 
-        model = load_model(filename, verbose=False)
+        model = load_model(model_name, verbose=False)
 
         if verbose:
             print("Transformation Pipeline and Model Successfully Loaded")