stage

AlexanderDokuchaev · AlexanderDokuchaev · commit 1efe638dded4 · 2024-12-30T16:58:02.000+02:00
diff --git a/nncf/tensorflow/sparsity/rb/algorithm.py b/nncf/tensorflow/sparsity/rb/algorithm.py
@@ -193,4 +193,4 @@ def disable_scheduler(self):
         self._scheduler = StubCompressionScheduler()
 
     def compression_stage(self) -> CompressionStage:
-        raise NotImplementedError("RB sparsity algorithm do not support compression stages")
+        raise None
