enforce abstract methods

vllm-project · Sep 12, 2024 · c962bbf · c962bbf
1 parent 5bfb497
commit c962bbf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/llmcompressor/modifiers/utils/compression_wrapper.py b/src/llmcompressor/modifiers/utils/compression_wrapper.py
@@ -1,4 +1,4 @@
-from abc import ABC
+from abc import ABC, abstractmethod
 from typing import Optional, Set
 
 import torch
@@ -72,12 +72,14 @@ def free(self):
         """
         delattr(self, "nsamples")
 
+    @abstractmethod
     def add_batch(self, *args, **kwargs):
         """
         Add a batch of layer input and output data to the layer statistics calculation
         """
         raise NotImplementedError("Child class must implement `add_batch`")
 
+    @abstractmethod
     def compress(self, *args, **kwargs):
         """
         Run pruning on the layer up to the target sparsity