avoid lambda for mp

rico_hdl/rico_hdl.py

@@ -142,6 +142,18 @@ def safetensor_generator_s1(patch_path: str) -> bytes:
     return save(data, metadata=None)
 
 
+def optional_compressed_safetensor_generator(
+    safetensor_generator, compress: bool = False
+):
+    if not compress:
+        return safetensor_generator
+
+    def wrapper(x):
+        return zstd_compressor(safetensor_generator(x))
+
+    return wrapper
+
+
 @app.command()
 def hyspecnet_11k(
     target_dir: TargetDir,
@@ -266,27 +278,22 @@ def bigearthnet(
     # Otherwise an error in the latter CLI argument could produce an incomplete LMDB
     env = open_lmdb(target_dir)
 
-    safetensor_gen_s1 = (
-        (lambda x: zstd_compressor(safetensor_generator_s1(x)))
-        if compress
-        else safetensor_generator_s1
-    )
-    safetensor_gen_s2 = (
-        (lambda x: zstd_compressor(safetensor_generator_s2(x)))
-        if compress
-        else safetensor_generator_s2
-    )
-
     if bigearthnet_s1_dir is not None:
         log.debug("Writing BigEarthNet-S1 data into LMDB")
         lmdb_writer(
-            env, s1_patch_paths, bigearthnet_lmdb_key_extractor, safetensor_gen_s1
+            env,
+            s1_patch_paths,
+            bigearthnet_lmdb_key_extractor,
+            optional_compressed_safetensor_generator(safetensor_generator_s1, compress),
         )
 
     if bigearthnet_s2_dir is not None:
         log.debug("Writing BigEarthNet-S2 data into LMDB")
         lmdb_writer(
-            env, s2_patch_paths, bigearthnet_lmdb_key_extractor, safetensor_gen_s2
+            env,
+            s2_patch_paths,
+            bigearthnet_lmdb_key_extractor,
+            optional_compressed_safetensor_generator(safetensor_generator_s2, compress),
         )