adds NRV to universal decompressor

binref · Apr 9, 2024 · 71179ae · 71179ae
1 parent db91e75
commit 71179ae
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/refinery/units/compression/decompress.py b/refinery/units/compression/decompress.py
@@ -5,20 +5,21 @@
 from refinery.units import Arg, Unit, RefineryPartialResult
 from refinery.lib.types import INF
 
-from .ap     import aplib  # noqa
-from .blz    import blz    # noqa
-from .bz2    import bz2    # noqa
-from .jcalg  import jcalg  # noqa
-from .lz     import lzma   # noqa
-from .lz4    import lz4    # noqa
-from .lzjb   import lzjb   # noqa
-from .lznt1  import lznt1  # noqa
-from .lzo    import lzo    # noqa
-from .szdd   import szdd   # noqa
-from .zl     import zl     # noqa
-from .qlz    import qlz    # noqa
-from .lzf    import lzf    # noqa
-from .lzw    import lzw    # noqa
+from .ap import aplib
+from .blz import blz
+from .bz2 import bz2
+from .jcalg import jcalg
+from .lz import lzma
+from .lz4 import lz4
+from .lzjb import lzjb
+from .lznt1 import lznt1
+from .lzo import lzo
+from .szdd import szdd
+from .zl import zl
+from .qlz import qlz
+from .lzf import lzf
+from .lzw import lzw
+from .nrv import nrv2b, nrv2d, nrv2e
 
 
 class decompress(Unit):
@@ -60,7 +61,7 @@ def __init__(
         )
         self.engines: List[Unit] = [
             engine.assemble() for engine in [
-                szdd, zl, lzma, aplib, qlz, lzf, lzw, jcalg, bz2, blz, lzjb, lz4, lzo, lznt1]
+                szdd, zl, lzma, aplib, qlz, lzf, lzw, jcalg, bz2, blz, lzjb, lz4, lzo, lznt1, nrv2e, nrv2d, nrv2b]
         ]
         for engine in self.engines:
             engine.log_detach()