"""Causality axis enum for specifying causal convolution dimensions.""" from enum import Enum class CausalityAxis(Enum): """Enum for specifying the causality axis in causal convolutions.""" NONE = None WIDTH = "width" HEIGHT = "height" WIDTH_COMPATIBILITY = "width-compatibility"