hotfix: disable dynamic intermediary skipping for top level

ZettaAI · Jan 6, 2024 · bc84d38 · bc84d38
1 parent eb3ff9e
commit bc84d38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zetta_utils/mazepa_layer_processing/common/subchunkable_apply_flow.py b/zetta_utils/mazepa_layer_processing/common/subchunkable_apply_flow.py
@@ -884,7 +884,7 @@ def _build_subchunkable_apply_flow(  # pylint: disable=keyword-arg-before-vararg
             ),
             allow_cache=(allow_cache_up_to_level >= level + 1),
             clear_cache_on_return=(allow_cache_up_to_level == level + 1),
-            force_intermediaries=not (skip_intermediaries) and ((level != num_levels - 1)),
+            force_intermediaries=not (skip_intermediaries),
         )
 
     return flow_schema(idx, dst, op_args, op_kwargs)