xobjects 0.5.5__tar.gz → 0.5.7__tar.gz

{xobjects-0.5.5/xobjects.egg-info → xobjects-0.5.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.5.5
+Version: 0.5.7
 Summary: In-memory serialization and code generator for CPU and GPU
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xobjects

xobjects-0.5.7/xobjects/_version.py

@@ -0,0 +1 @@
+__version__ = "0.5.7"

{xobjects-0.5.5 → xobjects-0.5.7}/xobjects/general.py

@@ -18,6 +18,13 @@ _print = Print()
 
 
 def assert_allclose(a, b, rtol=0, atol=0, max_outliers=0):
+
+    try:
+        if a == b: # if this passes we return without raising
+            return
+    except:
+        pass
+
     if hasattr(a, "get"):
         a = a.get()
     if hasattr(b, "get"):

{xobjects-0.5.5 → xobjects-0.5.7}/xobjects/hybrid_class.py

@@ -40,7 +40,6 @@ class _FieldOfDressed:
         if self.isnplikearray:
             self.__get__(container=container)[:] = value
         elif hasattr(value, "_xobject"):  # value is a dressed xobject
-
             # Copy xobject data from value inside self._xobject
             # (unless same memory area or Ref and same buffer,
             #  in the latter case reference mechanism is used)
@@ -390,14 +389,17 @@ class HybridClass(metaclass=MetaHybridClass):
         return self._xobject.compile_kernels(*args, **kwargs)
 
     def __repr__(self):
-
         if hasattr(self, "_repr_fields"):
             fnames = self._repr_fields
         else:
             fnames = []
             if hasattr(self, "_add_to_repr"):
                 fnames += self._add_to_repr
-            fnames += [fname for fname in self._fields]
+            fnames += [
+                fname
+                for fname in self._xofields.keys()
+                if not fname.startswith("_")
+            ]
             if hasattr(self, "_skip_in_repr"):
                 fnames = [ff for ff in fnames if ff not in self._skip_in_repr]
 
@@ -406,6 +408,12 @@ class HybridClass(metaclass=MetaHybridClass):
             vv = getattr(self, fname)
             if isinstance(vv, float):
                 vvrepr = f"{vv:.3g}"
+            elif isinstance(vv, np.ndarray):
+                vvrepr = repr(vv)
+                # kill "array("
+                vvrepr = vvrepr.split("(", 1)[1]
+                # kill trailing ")"
+                vvrepr = vvrepr.rsplit(")", 1)[0]
             else:
                 vvrepr = repr(vv)
             args.append(f"{fname}={vvrepr}")

{xobjects-0.5.5 → xobjects-0.5.7/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.5.5
+Version: 0.5.7
 Summary: In-memory serialization and code generator for CPU and GPU
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xobjects

xobjects-0.5.5/xobjects/_version.py

@@ -1 +0,0 @@
-__version__ = "0.5.5"