From 0dd37f97db13b0b5ae0c6c5a69d83191ad760aec Mon Sep 17 00:00:00 2001
From: nickfraser <icanlosh@gmail.com>
Date: Sat, 16 Dec 2023 14:09:54 +0000
Subject: [PATCH] Fix (graph): typo in class name (#765)

---
 src/brevitas/graph/base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/brevitas/graph/base.py b/src/brevitas/graph/base.py
index d594c851d..620312641 100644
--- 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 = {}