Fix Unit.create for setting base (#37)

chaobrain · Aug 10, 2024 · 3a870f2 · 3a870f2
1 parent 1abcb73
commit 3a870f2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/brainunit/_base.py b/brainunit/_base.py
@@ -1395,7 +1395,13 @@ def has_same_dim(self, other: 'Unit') -> bool:
     return get_dim(self) == other_dim
 
   @staticmethod
-  def create(dim: Dimension, name: str, dispname: str, scale: int = 0) -> 'Unit':
+  def create(
+      dim: Dimension,
+      name: str,
+      dispname: str,
+      scale: int = 0,
+      base: float = 10.
+  ) -> 'Unit':
     """
     Create a new named unit.
 
@@ -1419,6 +1425,7 @@ def create(dim: Dimension, name: str, dispname: str, scale: int = 0) -> 'Unit':
     u = Unit(
       dim=dim,
       scale=scale,
+      base=base,
       name=name,
       dispname=dispname,
     )