Changes to schema adjustment utilities.

PiperOrigin-RevId: 588347336

Author: jan-matthis

decorators/flow.py

@@ -18,7 +18,7 @@
 
 from connectomics.common import jax_utils
 # pylint: disable=g-importing-member
-from connectomics.volume.decorators import _adjust_schema_for_chunksize
+from connectomics.volume.decorators import adjust_schema_for_chunksize
 from connectomics.volume.decorators import Decorator
 from connectomics.volume.decorators import Filter
 # pylint: enable=g-importing-member
@@ -72,7 +72,7 @@ def filt_read(domain: ts.IndexDomain, array: np.ndarray,
 
     schema = input_ts.schema
     if self._min_chunksize is not None:
-      schema = _adjust_schema_for_chunksize(schema, self._min_chunksize)
+      schema = adjust_schema_for_chunksize(schema, self._min_chunksize)
 
     # Remove non-spatial dimensions from `fc`-dimension (0th).
     json = schema.to_json()