Bring your towel and join one of the fastest growing data communities. Welcome to our second-generation open source orchestration platform, a completely rethought approach to dataflow automation.

Prefect Community

Hi Prefect Folks - I am running Prefect 1.x and using Kubernetes Agents with Dask pods to execute tasks. I’ve had the need to create my own `StorageSchema`. Is there a recommended approach to ensuring my custom StorageSchema is globally accessible in the `StorageSchema.type_schemas`? As a workaround we are calling `StorageSchema.type_schemas["MyCustomStorageSchema"] = MyCustomStorageSchema` in our CLI wrapper before creating agents so that the custom schema is accessible, but when we create child flows via Dask workers the custom storage is not set and the flow fails.

This is how the type schemas are loaded
<https://github.com/PrefectHQ/prefect/blob/1.x/src/prefect/serialization/storage.py#L187>