More informative migration error

CovertLab · Dec 5, 2023 · a1472a7 · a1472a7
1 parent 07cdb8d
commit a1472a7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/migration/migration_utils.py b/migration/migration_utils.py
@@ -370,8 +370,9 @@ def recursive_compare(d1, d2, level='root', include_callable=False,
     elif isinstance(d1, np.ndarray) or isinstance(d2, np.ndarray):
         try:
             np.testing.assert_array_almost_equal_nulp(d1, d2)
-        except AssertionError:
-            print('{:<20} {} != {}'.format(level, d1, d2))
+        except AssertionError as e:
+            print('{:<20} {} != {}'.format(level, np.array(d1), np.array(d2)))
+            print(e)
             return False
         except TypeError:
             if not np.array_equal(d1, d2):