def storage(cls):
        if getenv("ENV") == "dev":
            # from prefect.storage import Local

            # return Local()
            from prefect.storage import Docker

            return Docker()
        else:
            from prefect.storage import Docker

            return Docker(
                registry_url=f"{Secret('AWS_ACCOUNT_ID').get()}.<http://DKR.ECR.US-EAST-1.AMAZONAWS.COM|DKR.ECR.US-EAST-1.AMAZONAWS.COM>",
                dockerfile="Dockerfile",
                image_name="platform-prefect",
                image_tag="latest",
                path="../",
            )

stored_as_script=True