Added pickle type to load

Author: christopherbunn

evalml/automl/automl_search.py

@@ -1337,11 +1337,14 @@ def save(
             pkl_lib.dump(self, f, protocol=pickle_protocol)
 
     @staticmethod
-    def load(file_path):
+    def load(
+        file_path,
+    ):
         """Loads AutoML object at file path
 
         Arguments:
             file_path (str): location to find file to load
+            pickle_type {"pickle", "cloudpickle"}: the pickling library to use. Currently not used since the standard pickle library can handle cloudpickles.
 
         Returns:
             AutoSearchBase object

evalml/tests/automl_tests/test_automl.py

@@ -665,7 +665,9 @@ def test_automl_serialization(pickle_type, X_y_binary, tmpdir):
     )
     automl.search()
     # Testing pickling of SearchIterationPlot object
-    automl.search_iteration_plot = automl.plot.search_iteration_plot(interactive_plot=True)
+    automl.search_iteration_plot = automl.plot.search_iteration_plot(
+        interactive_plot=True
+    )
     if pickle_type == "invalid":
         with pytest.raises(
             ValueError,