fix typo in code

thuml · Sep 7, 2024 · bad9a0f · bad9a0f
1 parent 9606941
commit bad9a0f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/depyf/optimization.py b/depyf/optimization.py
@@ -8,7 +8,7 @@
 import torch
 
 
-class TorchCompileWrapperWithCustomDispacther:
+class TorchCompileWrapperWithCustomDispatcher:
     """
     A wrapper class for torch.compile, with a custom dispatch logic.
     Subclasses should:

diff --git a/tests/test_pytorch/test_wrapper.py b/tests/test_pytorch/test_wrapper.py
@@ -2,7 +2,7 @@
 
 import torch
 
-from depyf.optimization import TorchCompileWrapperWithCustomDispacther
+from depyf.optimization import TorchCompileWrapperWithCustomDispatcher
 
 
 class MyMod(torch.nn.Module):
@@ -13,7 +13,7 @@ def forward(self, x: torch.Tensor, cache: Optional[torch.Tensor] = None):
         return x * 2
 
 
-class MyWrapper(TorchCompileWrapperWithCustomDispacther):
+class MyWrapper(TorchCompileWrapperWithCustomDispatcher):
 
     def __init__(self, model):
         self.model = model