BUG: fix caching error with UnitRegistry.modify

yt-project · Nov 11, 2023 · 2aee0fc · 2aee0fc
1 parent 99cf350
commit 2aee0fc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/unyt/unit_registry.py b/unyt/unit_registry.py
@@ -216,6 +216,8 @@ def modify(self, symbol, base_value):
             new_dimensions = self.lut[symbol][1]
 
         self.lut[symbol] = (float(base_value), new_dimensions) + self.lut[symbol][2:]
+        if symbol in self._unit_object_cache:
+            self._unit_object_cache.pop(symbol)
 
     def keys(self):
         """