Access model environments

Model environments help configure behavior based on deployment stage (e.g., production vs. staging). You can access the environment details via kwargs in the Model class.

​

1. Retrieve Environment Variables

Access the environment in __init__:

def __init__(self, **kwargs):
    self._environment = kwargs["environment"]

​

2. Configure Behavior Based on Environment

Use environment variables in the load function:

def load(self):
    if self._environment.get("name") == "production":
        # Production setup
        self.setup_sentry()
        self.setup_logging(level="INFO")
        self.load_production_weights()
    else:
        # Default setup for staging or development deployments
        self.setup_logging(level="DEBUG")
        self.load_default_weights()

Why use this?

• Customize logging levels
• Load environment-specific model weights
• Enable monitoring tools (e.g., Sentry)