bugfix

binref · Dec 3, 2024 · 79d2f19 · 79d2f19
1 parent 3204b47
commit 79d2f19
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/refinery/units/compression/zstd.py b/refinery/units/compression/zstd.py
@@ -13,12 +13,12 @@ def _pyzstd():
         return pyzstd
 
     def process(self, data):
-        return self._pyzstd.ZstdDecompressor().decompress(data)
+        zd = self._pyzstd.ZstdDecompressor()
+        return zd.decompress(data)
 
     def reverse(self, data):
         zc = self._pyzstd.ZstdCompressor()
-        zc.compress(data)
-        return zc.flush()
+        return zc.compress(data) + zc.flush()
 
     @classmethod
     def handles(self, data: bytearray) -> bool: