Workaround for xsuite/xobjects#137: don't use <64b fields in elements

xsuite · Jul 11, 2024 · b3e0f9b · b3e0f9b
1 parent e7ac347
commit b3e0f9b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/xtrack/beam_elements/elements.py b/xtrack/beam_elements/elements.py
@@ -990,8 +990,8 @@ class Sextupole(BeamElement):
         'inv_factorial_order': xo.Float64,
         'knl': xo.Float64[ALLOCATED_MULTIPOLE_ORDER + 1],
         'ksl': xo.Float64[ALLOCATED_MULTIPOLE_ORDER + 1],
-        'edge_entry_active': xo.Field(xo.UInt8, default=False),
-        'edge_exit_active': xo.Field(xo.UInt8, default=False),
+        'edge_entry_active': xo.Field(xo.UInt64, default=False),
+        'edge_exit_active': xo.Field(xo.UInt64, default=False),
     }
 
     _skip_in_to_dict = ['_order', 'inv_factorial_order']  # defined by knl, etc.
@@ -1091,8 +1091,8 @@ class Octupole(BeamElement):
         'inv_factorial_order': xo.Float64,
         'knl': xo.Float64[ALLOCATED_MULTIPOLE_ORDER + 1],
         'ksl': xo.Float64[ALLOCATED_MULTIPOLE_ORDER + 1],
-        'edge_entry_active': xo.Field(xo.UInt8, default=False),
-        'edge_exit_active': xo.Field(xo.UInt8, default=False),
+        'edge_entry_active': xo.Field(xo.UInt64, default=False),
+        'edge_exit_active': xo.Field(xo.UInt64, default=False),
     }
 
     _skip_in_to_dict = ['_order', 'inv_factorial_order']  # defined by knl, etc.
@@ -1191,8 +1191,8 @@ class Quadrupole(BeamElement):
         'inv_factorial_order': xo.Float64,
         'knl': xo.Float64[ALLOCATED_MULTIPOLE_ORDER + 1],
         'ksl': xo.Float64[ALLOCATED_MULTIPOLE_ORDER + 1],
-        'edge_entry_active': xo.Field(xo.UInt8, default=False),
-        'edge_exit_active': xo.Field(xo.UInt8, default=False),
+        'edge_entry_active': xo.Field(xo.UInt64, default=False),
+        'edge_exit_active': xo.Field(xo.UInt64, default=False),
     }
 
     _skip_in_to_dict = ['_order', 'inv_factorial_order']  # defined by knl, etc.