diff --git a/chi/container.py b/chi/container.py index 2d8785f..dcb4ae9 100644 --- a/chi/container.py +++ b/chi/container.py @@ -17,7 +17,7 @@ import os import tarfile import time -from typing import List, Optional, Tuple +from typing import Dictionary, List, Optional, Tuple from IPython.display import HTML, display from packaging.version import Version @@ -62,6 +62,8 @@ class Container: id (str): The ID of the container. created_at (str): The timestamp when the container was created. status (str): The current status of the container. + environment (Dictionary[str, str]): A dictionary of environment variables for the container. + device_profiles (List[str]): A list of device profiles to be configured on the container. """ def __init__( @@ -75,6 +77,8 @@ def __init__( runtime: str = None, command: List[str] = None, workdir: str = None, + environment: Dictionary[str, str] = {}, + device_profiles: List[str] = [], ): self.name = name self.image_ref = image_ref @@ -88,6 +92,8 @@ def __init__( self._status = None self.command = command self.workdir = workdir + self.environment = environment + self.device_profiles = device_profiles @classmethod def from_zun_container(cls, zun_container): @@ -152,6 +158,8 @@ def submit( start=self.start, start_timeout=self.start_timeout, runtime=self.runtime, + environment=self.environment, + device_profiles=self.device_profiles, **kwargs, )