Add coil names and types

.pre-commit-config.yaml

@@ -12,10 +12,10 @@ repos:
         exclude: test_data
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.4.8
+    rev: v0.5.0
     hooks:
       - id: ruff
-        args: [--fix, --exit-non-zero-on-fix]
+        args: [--fix]
       - id: ruff-format
 
   - repo: https://github.com/pre-commit/mirrors-prettier

eqdsk/cocos.py

@@ -352,7 +352,7 @@ def identify_cocos(
         raise ValueError(
             "The sign of qpsi is not consistent across the flux surfaces.",
         )
-    sign_q = Sign(sign_q.max())
+    sign_q = Sign(int(sign_q.max()) or 1)
 
     sign_Bp = sign_Ip * sign_psi_inc_towards_boundary
     sign_rho_theta_phi = sign_Ip * sign_B0 * sign_q

eqdsk/file.py

@@ -122,6 +122,10 @@ class EQDSKInterface:
     """Safety factor values on the 1-D flux grid [dimensionless]."""
     file_name: str | None = None
     """The EQDSK file the data originates from."""
+    coil_names: list[str] | None = None
+    """Name of the coils"""
+    coil_types: list[str] | None = None
+    """Type of the coils"""
 
     def __post_init__(self):
         """Calculate derived parameters if they're not given."""
@@ -332,7 +336,7 @@ def _read_json(file_path: Path) -> dict[str, Any]:
         data = json.load(file)
 
     for k, value in data.items():
-        if isinstance(value, list):
+        if isinstance(value, list) and k not in {"coil_type", "coil_names"}:
             data[k] = np.asarray(value)
 
     # For backward compatibility where 'psinorm' was sometimes 'pnorm'