diff --git a/environments/py-3.10-linux-64-dev.conda.lock.yml b/environments/py-3.10-linux-64-dev.conda.lock.yml
index 547f37dc..c280eacb 100644
--- a/environments/py-3.10-linux-64-dev.conda.lock.yml
+++ b/environments/py-3.10-linux-64-dev.conda.lock.yml
@@ -197,8 +197,8 @@ dependencies:
   - ptyprocess=0.7.0=pyhd8ed1ab_1
   - pure_eval=0.2.3=pyhd8ed1ab_1
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py310hd8f68c5_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py310hd8f68c5_0
   - pydiso=0.1.2=py310h4b187eb_1
   - pygments=2.19.2=pyhd8ed1ab_0
   - pylint=4.0.2=pyhcf101f3_0
diff --git a/environments/py-3.10-linux-64.conda.lock.yml b/environments/py-3.10-linux-64.conda.lock.yml
index d216c7e3..0961b70c 100644
--- a/environments/py-3.10-linux-64.conda.lock.yml
+++ b/environments/py-3.10-linux-64.conda.lock.yml
@@ -113,8 +113,8 @@ dependencies:
   - psutil=7.1.3=py310h139afa4_0
   - pthread-stubs=0.4=hb9d3cd8_1002
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py310hd8f68c5_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py310hd8f68c5_0
   - pydiso=0.1.2=py310h4b187eb_1
   - pymatsolver=0.3.1=pyh48887ae_201
   - pyparsing=3.2.5=pyhcf101f3_0
diff --git a/environments/py-3.10-win-64-dev.conda.lock.yml b/environments/py-3.10-win-64-dev.conda.lock.yml
index a27a96ad..6f2e2cee 100644
--- a/environments/py-3.10-win-64-dev.conda.lock.yml
+++ b/environments/py-3.10-win-64-dev.conda.lock.yml
@@ -177,8 +177,8 @@ dependencies:
   - pthread-stubs=0.4=h0e40799_1002
   - pure_eval=0.2.3=pyhd8ed1ab_1
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py310h034784e_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py310h034784e_0
   - pydiso=0.1.2=py310h3dbbb0c_1
   - pygments=2.19.2=pyhd8ed1ab_0
   - pylint=4.0.2=pyhcf101f3_0
diff --git a/environments/py-3.10-win-64.conda.lock.yml b/environments/py-3.10-win-64.conda.lock.yml
index cf02d45d..83a3b25d 100644
--- a/environments/py-3.10-win-64.conda.lock.yml
+++ b/environments/py-3.10-win-64.conda.lock.yml
@@ -96,8 +96,8 @@ dependencies:
   - psutil=7.1.3=py310h1637853_0
   - pthread-stubs=0.4=h0e40799_1002
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py310h034784e_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py310h034784e_0
   - pydiso=0.1.2=py310h3dbbb0c_1
   - pymatsolver=0.3.1=pyh48887ae_201
   - pyparsing=3.2.5=pyhcf101f3_0
diff --git a/environments/py-3.11-linux-64-dev.conda.lock.yml b/environments/py-3.11-linux-64-dev.conda.lock.yml
index 14a4af3d..6ad38454 100644
--- a/environments/py-3.11-linux-64-dev.conda.lock.yml
+++ b/environments/py-3.11-linux-64-dev.conda.lock.yml
@@ -198,8 +198,8 @@ dependencies:
   - ptyprocess=0.7.0=pyhd8ed1ab_1
   - pure_eval=0.2.3=pyhd8ed1ab_1
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py311h902ca64_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py311h902ca64_0
   - pydiso=0.1.2=py311h6070e36_1
   - pygments=2.19.2=pyhd8ed1ab_0
   - pylint=4.0.2=pyhcf101f3_0
diff --git a/environments/py-3.11-linux-64.conda.lock.yml b/environments/py-3.11-linux-64.conda.lock.yml
index fa047881..2443e3b3 100644
--- a/environments/py-3.11-linux-64.conda.lock.yml
+++ b/environments/py-3.11-linux-64.conda.lock.yml
@@ -114,8 +114,8 @@ dependencies:
   - psutil=7.1.3=py311haee01d2_0
   - pthread-stubs=0.4=hb9d3cd8_1002
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py311h902ca64_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py311h902ca64_0
   - pydiso=0.1.2=py311h6070e36_1
   - pymatsolver=0.3.1=pyh48887ae_201
   - pyparsing=3.2.5=pyhcf101f3_0
diff --git a/environments/py-3.11-win-64-dev.conda.lock.yml b/environments/py-3.11-win-64-dev.conda.lock.yml
index 2385b550..5a62ee14 100644
--- a/environments/py-3.11-win-64-dev.conda.lock.yml
+++ b/environments/py-3.11-win-64-dev.conda.lock.yml
@@ -178,8 +178,8 @@ dependencies:
   - pthread-stubs=0.4=h0e40799_1002
   - pure_eval=0.2.3=pyhd8ed1ab_1
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py311hf51aa87_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py311hf51aa87_0
   - pydiso=0.1.2=py311h1c8cef1_1
   - pygments=2.19.2=pyhd8ed1ab_0
   - pylint=4.0.2=pyhcf101f3_0
diff --git a/environments/py-3.11-win-64.conda.lock.yml b/environments/py-3.11-win-64.conda.lock.yml
index 4e6b191e..d350f589 100644
--- a/environments/py-3.11-win-64.conda.lock.yml
+++ b/environments/py-3.11-win-64.conda.lock.yml
@@ -97,8 +97,8 @@ dependencies:
   - psutil=7.1.3=py311hf893f09_0
   - pthread-stubs=0.4=h0e40799_1002
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py311hf51aa87_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py311hf51aa87_0
   - pydiso=0.1.2=py311h1c8cef1_1
   - pymatsolver=0.3.1=pyh48887ae_201
   - pyparsing=3.2.5=pyhcf101f3_0
diff --git a/environments/py-3.12-linux-64-dev.conda.lock.yml b/environments/py-3.12-linux-64-dev.conda.lock.yml
index 31c17b64..e6a75f2e 100644
--- a/environments/py-3.12-linux-64-dev.conda.lock.yml
+++ b/environments/py-3.12-linux-64-dev.conda.lock.yml
@@ -200,8 +200,8 @@ dependencies:
   - ptyprocess=0.7.0=pyhd8ed1ab_1
   - pure_eval=0.2.3=pyhd8ed1ab_1
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py312h868fb18_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py312h868fb18_0
   - pydiso=0.1.2=py312h686354e_1
   - pygments=2.19.2=pyhd8ed1ab_0
   - pylint=4.0.2=pyhcf101f3_0
diff --git a/environments/py-3.12-linux-64.conda.lock.yml b/environments/py-3.12-linux-64.conda.lock.yml
index df866cbe..1751176b 100644
--- a/environments/py-3.12-linux-64.conda.lock.yml
+++ b/environments/py-3.12-linux-64.conda.lock.yml
@@ -114,8 +114,8 @@ dependencies:
   - psutil=7.1.3=py312h5253ce2_0
   - pthread-stubs=0.4=hb9d3cd8_1002
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py312h868fb18_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py312h868fb18_0
   - pydiso=0.1.2=py312h686354e_1
   - pymatsolver=0.3.1=pyh48887ae_201
   - pyparsing=3.2.5=pyhcf101f3_0
diff --git a/environments/py-3.12-win-64-dev.conda.lock.yml b/environments/py-3.12-win-64-dev.conda.lock.yml
index 53ff3062..f85a1b59 100644
--- a/environments/py-3.12-win-64-dev.conda.lock.yml
+++ b/environments/py-3.12-win-64-dev.conda.lock.yml
@@ -180,8 +180,8 @@ dependencies:
   - pthread-stubs=0.4=h0e40799_1002
   - pure_eval=0.2.3=pyhd8ed1ab_1
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py312hdabe01f_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py312hdabe01f_0
   - pydiso=0.1.2=py312h3fe0e52_1
   - pygments=2.19.2=pyhd8ed1ab_0
   - pylint=4.0.2=pyhcf101f3_0
diff --git a/environments/py-3.12-win-64.conda.lock.yml b/environments/py-3.12-win-64.conda.lock.yml
index caf01261..6f2accee 100644
--- a/environments/py-3.12-win-64.conda.lock.yml
+++ b/environments/py-3.12-win-64.conda.lock.yml
@@ -97,8 +97,8 @@ dependencies:
   - psutil=7.1.3=py312he5662c2_0
   - pthread-stubs=0.4=h0e40799_1002
   - pycparser=2.22=pyh29332c3_1
-  - pydantic=2.12.3=pyh3cfb1c2_0
-  - pydantic-core=2.41.4=py312hdabe01f_0
+  - pydantic=2.12.4=pyh3cfb1c2_0
+  - pydantic-core=2.41.5=py312hdabe01f_0
   - pydiso=0.1.2=py312h3fe0e52_1
   - pymatsolver=0.3.1=pyh48887ae_201
   - pyparsing=3.2.5=pyhcf101f3_0
diff --git a/py-3.10.conda-lock.yml b/py-3.10.conda-lock.yml
index 5c00eb7f..2edecd85 100644
--- a/py-3.10.conda-lock.yml
+++ b/py-3.10.conda-lock.yml
@@ -5346,41 +5346,41 @@ package:
   category: main
   optional: false
 - name: pydantic
-  version: 2.12.3
+  version: 2.12.4
   manager: conda
   platform: linux-64
   dependencies:
     annotated-types: '>=0.6.0'
-    pydantic-core: 2.41.4
+    pydantic-core: 2.41.5
     python: '>=3.10'
     typing-extensions: '>=4.6.1'
     typing-inspection: '>=0.4.2'
     typing_extensions: '>=4.14.1'
-  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.3-pyh3cfb1c2_0.conda
+  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.4-pyh3cfb1c2_0.conda
   hash:
-    md5: 2cb5d62fdf68deb0263663598feb9fc5
-    sha256: 6a940747e8445653224dcff95fadf1060c66b9e544fdb0ed469b70a98c3aee7e
+    md5: bf6ce72315b6759453d8c90a894e9e4c
+    sha256: c51297f0f6ef13776cc5b61c37d00c0d45faaed34f81d196e64bebc989f3e497
   category: main
   optional: false
 - name: pydantic
-  version: 2.12.3
+  version: 2.12.4
   manager: conda
   platform: win-64
   dependencies:
     annotated-types: '>=0.6.0'
-    pydantic-core: 2.41.4
+    pydantic-core: 2.41.5
     python: '>=3.10'
     typing-extensions: '>=4.6.1'
     typing-inspection: '>=0.4.2'
     typing_extensions: '>=4.14.1'
-  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.3-pyh3cfb1c2_0.conda
+  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.4-pyh3cfb1c2_0.conda
   hash:
-    md5: 2cb5d62fdf68deb0263663598feb9fc5
-    sha256: 6a940747e8445653224dcff95fadf1060c66b9e544fdb0ed469b70a98c3aee7e
+    md5: bf6ce72315b6759453d8c90a894e9e4c
+    sha256: c51297f0f6ef13776cc5b61c37d00c0d45faaed34f81d196e64bebc989f3e497
   category: main
   optional: false
 - name: pydantic-core
-  version: 2.41.4
+  version: 2.41.5
   manager: conda
   platform: linux-64
   dependencies:
@@ -5389,14 +5389,14 @@ package:
     python: ''
     python_abi: 3.10.*
     typing-extensions: '>=4.6.0,!=4.7.0'
-  url: https://repo.prefix.dev/conda-forge/linux-64/pydantic-core-2.41.4-py310hd8f68c5_0.conda
+  url: https://repo.prefix.dev/conda-forge/linux-64/pydantic-core-2.41.5-py310hd8f68c5_0.conda
   hash:
-    md5: e6a2c9696ff99ae8a47c898f643cc2b0
-    sha256: 5108adc2d3c25343cfc26f8fccdb407427eb7053e4fb4b37f07f1c95187fd6cc
+    md5: 9b2338c7021dca0c0f3149e6b1e11bdb
+    sha256: 37663ebef306620b84e9c3f26bcb122f7f70904374c1fa305221cc4d87aae403
   category: main
   optional: false
 - name: pydantic-core
-  version: 2.41.4
+  version: 2.41.5
   manager: conda
   platform: win-64
   dependencies:
@@ -5406,10 +5406,10 @@ package:
     ucrt: '>=10.0.20348.0'
     vc: '>=14.3,<15'
     vc14_runtime: '>=14.44.35208'
-  url: https://repo.prefix.dev/conda-forge/win-64/pydantic-core-2.41.4-py310h034784e_0.conda
+  url: https://repo.prefix.dev/conda-forge/win-64/pydantic-core-2.41.5-py310h034784e_0.conda
   hash:
-    md5: 63b4a7922606d8e280d58c51380774bd
-    sha256: 51e5a7d247832664da04201d4e8a1750c361f0fd175f5e24c993a579f6bf0d6c
+    md5: 082a265124fef1094d6d7941441cdc15
+    sha256: f6eb0c918e7104b7f27c6f9589c7e2b0c1b73544f8f59c1cbba9ab73ec95df6e
   category: main
   optional: false
 - name: pydiso
diff --git a/py-3.11.conda-lock.yml b/py-3.11.conda-lock.yml
index e570ce08..c2c56f89 100644
--- a/py-3.11.conda-lock.yml
+++ b/py-3.11.conda-lock.yml
@@ -5372,41 +5372,41 @@ package:
   category: main
   optional: false
 - name: pydantic
-  version: 2.12.3
+  version: 2.12.4
   manager: conda
   platform: linux-64
   dependencies:
     annotated-types: '>=0.6.0'
-    pydantic-core: 2.41.4
+    pydantic-core: 2.41.5
     python: '>=3.10'
     typing-extensions: '>=4.6.1'
     typing-inspection: '>=0.4.2'
     typing_extensions: '>=4.14.1'
-  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.3-pyh3cfb1c2_0.conda
+  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.4-pyh3cfb1c2_0.conda
   hash:
-    md5: 2cb5d62fdf68deb0263663598feb9fc5
-    sha256: 6a940747e8445653224dcff95fadf1060c66b9e544fdb0ed469b70a98c3aee7e
+    md5: bf6ce72315b6759453d8c90a894e9e4c
+    sha256: c51297f0f6ef13776cc5b61c37d00c0d45faaed34f81d196e64bebc989f3e497
   category: main
   optional: false
 - name: pydantic
-  version: 2.12.3
+  version: 2.12.4
   manager: conda
   platform: win-64
   dependencies:
     annotated-types: '>=0.6.0'
-    pydantic-core: 2.41.4
+    pydantic-core: 2.41.5
     python: '>=3.10'
     typing-extensions: '>=4.6.1'
     typing-inspection: '>=0.4.2'
     typing_extensions: '>=4.14.1'
-  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.3-pyh3cfb1c2_0.conda
+  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.4-pyh3cfb1c2_0.conda
   hash:
-    md5: 2cb5d62fdf68deb0263663598feb9fc5
-    sha256: 6a940747e8445653224dcff95fadf1060c66b9e544fdb0ed469b70a98c3aee7e
+    md5: bf6ce72315b6759453d8c90a894e9e4c
+    sha256: c51297f0f6ef13776cc5b61c37d00c0d45faaed34f81d196e64bebc989f3e497
   category: main
   optional: false
 - name: pydantic-core
-  version: 2.41.4
+  version: 2.41.5
   manager: conda
   platform: linux-64
   dependencies:
@@ -5415,14 +5415,14 @@ package:
     python: ''
     python_abi: 3.11.*
     typing-extensions: '>=4.6.0,!=4.7.0'
-  url: https://repo.prefix.dev/conda-forge/linux-64/pydantic-core-2.41.4-py311h902ca64_0.conda
+  url: https://repo.prefix.dev/conda-forge/linux-64/pydantic-core-2.41.5-py311h902ca64_0.conda
   hash:
-    md5: a440367c3ba0b93cfb7c6b1b93989720
-    sha256: 7f7472dfc6a7275ea6c03b7fa7c1060fa5a3d0e39bbab16eac907d13421abd0e
+    md5: ffddb15810f766cdde04fe1af24d7168
+    sha256: 2632d271a937fe1aa932f40b282e8786c7d59158268042f158a31ef6bcc7edfe
   category: main
   optional: false
 - name: pydantic-core
-  version: 2.41.4
+  version: 2.41.5
   manager: conda
   platform: win-64
   dependencies:
@@ -5432,10 +5432,10 @@ package:
     ucrt: '>=10.0.20348.0'
     vc: '>=14.3,<15'
     vc14_runtime: '>=14.44.35208'
-  url: https://repo.prefix.dev/conda-forge/win-64/pydantic-core-2.41.4-py311hf51aa87_0.conda
+  url: https://repo.prefix.dev/conda-forge/win-64/pydantic-core-2.41.5-py311hf51aa87_0.conda
   hash:
-    md5: 67883b7e8b7260fa4571304e60cf0b01
-    sha256: 9ed29654119c308889d3cdc08ddbefa4b070cba554d7f7db1cf372325dc8bed5
+    md5: 7e97287ec2a853ff9588481dc19bc3ec
+    sha256: b99307c2f938550abd8e1ed2f1f5f57a81dac99f0c7daf86ccc2d8ad8a7e0e5c
   category: main
   optional: false
 - name: pydiso
diff --git a/py-3.12.conda-lock.yml b/py-3.12.conda-lock.yml
index 797faf05..cc8efe6f 100644
--- a/py-3.12.conda-lock.yml
+++ b/py-3.12.conda-lock.yml
@@ -5424,41 +5424,41 @@ package:
   category: main
   optional: false
 - name: pydantic
-  version: 2.12.3
+  version: 2.12.4
   manager: conda
   platform: linux-64
   dependencies:
     annotated-types: '>=0.6.0'
-    pydantic-core: 2.41.4
+    pydantic-core: 2.41.5
     python: '>=3.10'
     typing-extensions: '>=4.6.1'
     typing-inspection: '>=0.4.2'
     typing_extensions: '>=4.14.1'
-  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.3-pyh3cfb1c2_0.conda
+  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.4-pyh3cfb1c2_0.conda
   hash:
-    md5: 2cb5d62fdf68deb0263663598feb9fc5
-    sha256: 6a940747e8445653224dcff95fadf1060c66b9e544fdb0ed469b70a98c3aee7e
+    md5: bf6ce72315b6759453d8c90a894e9e4c
+    sha256: c51297f0f6ef13776cc5b61c37d00c0d45faaed34f81d196e64bebc989f3e497
   category: main
   optional: false
 - name: pydantic
-  version: 2.12.3
+  version: 2.12.4
   manager: conda
   platform: win-64
   dependencies:
     annotated-types: '>=0.6.0'
-    pydantic-core: 2.41.4
+    pydantic-core: 2.41.5
     python: '>=3.10'
     typing-extensions: '>=4.6.1'
     typing-inspection: '>=0.4.2'
     typing_extensions: '>=4.14.1'
-  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.3-pyh3cfb1c2_0.conda
+  url: https://repo.prefix.dev/conda-forge/noarch/pydantic-2.12.4-pyh3cfb1c2_0.conda
   hash:
-    md5: 2cb5d62fdf68deb0263663598feb9fc5
-    sha256: 6a940747e8445653224dcff95fadf1060c66b9e544fdb0ed469b70a98c3aee7e
+    md5: bf6ce72315b6759453d8c90a894e9e4c
+    sha256: c51297f0f6ef13776cc5b61c37d00c0d45faaed34f81d196e64bebc989f3e497
   category: main
   optional: false
 - name: pydantic-core
-  version: 2.41.4
+  version: 2.41.5
   manager: conda
   platform: linux-64
   dependencies:
@@ -5467,14 +5467,14 @@ package:
     python: ''
     python_abi: 3.12.*
     typing-extensions: '>=4.6.0,!=4.7.0'
-  url: https://repo.prefix.dev/conda-forge/linux-64/pydantic-core-2.41.4-py312h868fb18_0.conda
+  url: https://repo.prefix.dev/conda-forge/linux-64/pydantic-core-2.41.5-py312h868fb18_0.conda
   hash:
-    md5: e0767518fa45df8b484421a405f764c6
-    sha256: c0bcd8b16188ecb0b9148701d166888a91103e898c2401a45b290e8483d5bbca
+    md5: c7017a8aaec0ee5a5212d004e73bfdef
+    sha256: d1b924c342f3b526bb0a0d74e7ef8a8294f9df71355f0bbc619f9d3486931fe6
   category: main
   optional: false
 - name: pydantic-core
-  version: 2.41.4
+  version: 2.41.5
   manager: conda
   platform: win-64
   dependencies:
@@ -5484,10 +5484,10 @@ package:
     ucrt: '>=10.0.20348.0'
     vc: '>=14.3,<15'
     vc14_runtime: '>=14.44.35208'
-  url: https://repo.prefix.dev/conda-forge/win-64/pydantic-core-2.41.4-py312hdabe01f_0.conda
+  url: https://repo.prefix.dev/conda-forge/win-64/pydantic-core-2.41.5-py312hdabe01f_0.conda
   hash:
-    md5: fa35bb0b9c3dca0da27c46ed61a5dfd4
-    sha256: ee389c910c4fb3329c0dc36f3d68212c759c6082e7fa03d3d2ca85c2899b0a6b
+    md5: 79baaa876cfe843b62b0bcc3d0a1db2e
+    sha256: 931c28d0a67d4c0a388f172ffc1ee956a32a386fa5490b643406a1e02e8135d9
   category: main
   optional: false
 - name: pydiso
diff --git a/simpeg_drivers/driver.py b/simpeg_drivers/driver.py
index c4252c9c..f43ab7cb 100644
--- a/simpeg_drivers/driver.py
+++ b/simpeg_drivers/driver.py
@@ -26,6 +26,8 @@
 from pathlib import Path
 from time import time
 
+from typing_extensions import Self
+
 import numpy as np
 from dask import config as dconf
 
@@ -738,7 +740,7 @@ def configure_dask(self):
             dconf.set(scheduler="threads", pool=ThreadPool(n_cpu))
 
     @classmethod
-    def start(cls, filepath: str | Path | InputFile, **kwargs) -> BaseDriver:
+    def start(cls, filepath: str | Path | InputFile, **kwargs) -> Self:
         """
         Start the inversion driver.