[graph] Fix typo in class name

Xilinx · Nov 22, 2023 · b9ea062 · b9ea062
1 parent bd46f89
commit b9ea062
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/brevitas/graph/base.py b/src/brevitas/graph/base.py
@@ -17,7 +17,7 @@
 
 __all__ = [
     'Transform',
-    'PerInputTrasform',
+    'PerInputTransform',
     'GraphTransform',
     'PerInputModuleToModuleByHook',
     'ModuleToModule',
@@ -40,7 +40,7 @@ def apply(self, model: Module) -> Module:
         pass
 
 
-class PerInputTrasform(ABC):
+class PerInputTransform(ABC):
 
     @abstractmethod
     def apply(self, model: Module, inp: torch.Tensor) -> Module:
@@ -66,7 +66,7 @@ def apply(self, graph_model: GraphModule) -> GraphModule:
         return graph_model
 
 
-class PerInputModuleToModuleByHook(PerInputTrasform, ABC):
+class PerInputModuleToModuleByHook(PerInputTransform, ABC):
 
     def __init__(self):
         self.input_size_map = {}