revert change to fix failing tests

alteryx · Nov 17, 2021 · a06865b · a06865b
1 parent 5351665
commit a06865b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/evalml/pipelines/pipeline_base.py b/evalml/pipelines/pipeline_base.py
@@ -450,8 +450,8 @@ def graph_json(self):
                     param_dict[f"{param}_name"] = val.name
                     param_dict[f"{param}_parameters"] = val.parameters
                 else:
-                    if np.issubdtype(type(val), np.number):
-                        val = val.item()
+                    if isinstance(val, np.int64):
+                        val = int(val)
                     param_dict[param] = val
             nodes[comp_] = {"Parameters": param_dict, "Name": att_.name}