exclude MagnitudeMatrix from subclasses

Jaded-Encoding-Thaumaturgy · Nov 10, 2024 · 2c84d49 · 2c84d49
1 parent 6f1b5ff
commit 2c84d49
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vsmasktools/edge/_abstract.py b/vsmasktools/edge/_abstract.py
@@ -396,7 +396,7 @@ def _all_subclasses(cls: type[EdgeDetect] = EdgeDetect) -> set[type[EdgeDetect]]
     all_subclasses = {
         s for s in _all_subclasses()
         if s.__name__ not in {
-            'MatrixEdgeDetect', 'RidgeDetect', 'SingleMatrix', 'EuclideanDistance', 'Max',
+            'MatrixEdgeDetect', 'RidgeDetect', 'SingleMatrix', 'EuclideanDistance', 'MagnitudeMatrix', 'Max',
             'Matrix1D', 'SavitzkyGolay', 'SavitzkyGolayNormalise',
             'Matrix2x2', 'Matrix3x3', 'Matrix5x5'
         }
@@ -428,7 +428,7 @@ def _all_subclasses(cls: type[RidgeDetect] = RidgeDetect) -> set[type[RidgeDetec
     all_subclasses = {
         s for s in _all_subclasses()
         if s.__name__ not in {
-            'MatrixEdgeDetect', 'RidgeDetect', 'SingleMatrix', 'EuclideanDistance', 'Max',
+            'MatrixEdgeDetect', 'RidgeDetect', 'SingleMatrix', 'EuclideanDistance', 'MagnitudeMatrix', 'Max',
             'Matrix1D', 'SavitzkyGolay', 'SavitzkyGolayNormalise',
             'Matrix2x2', 'Matrix3x3', 'Matrix5x5'
         }