Merge pull request #1198 from mikel-brostrom/cmc_abc

cmc inherits abc
mikel-brostrom · Nov 22, 2023 · e4b02a1 · e4b02a1
2 parents 0733f9a + 8c077ce
commit e4b02a1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/boxmot/motion/cmc/cmc_interface.py b/boxmot/motion/cmc/cmc_interface.py
@@ -1,12 +1,14 @@
 # Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license
 
 import numpy as np
+from abc import ABC, abstractmethod
 
 
-class CMCInterface:
+class CMCInterface(ABC):
 
+    @abstractmethod
     def apply(self, im):
-        raise NotImplementedError('Subclasses must implement this method.')
+        pass
 
     def generate_mask(self, img, dets, scale):
         h, w = img.shape