Update Container interface

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,
         )