fix model generator

maxim-mityutko · Oct 8, 2024 · 4a1d375 · 4a1d375
1 parent 0a0e3a2
commit 4a1d375
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/brickflow/bundles/model.py b/brickflow/bundles/model.py
@@ -292,7 +292,7 @@ class Config:
     )
 
 
-class JobsSpec(BaseModel):
+class JobsEnvironmentsSpec(BaseModel):
     class Config:
         extra = "forbid"
         protected_namespaces = ()
@@ -307,15 +307,15 @@ class Config:
     )
 
 
-class Jobs(BaseModel):
+class JobsEnvironments(BaseModel):
     class Config:
         extra = "forbid"
         protected_namespaces = ()
 
     environment_key: str = Field(
         ..., description='The key of an environment. It has to be unique within a job.'
     )
-    spec: Optional[JobsSpec] = None
+    spec: Optional[JobsEnvironmentsSpec] = None
 
 
 class JobsGitSourceGitSnapshot(BaseModel):
@@ -2541,7 +2541,7 @@ class Config:
         description='Edit mode of the job.\n\n* `UI_LOCKED`: The job is in a locked UI state and cannot be modified.\n* `EDITABLE`: The job is in an editable state and can be modified.',
     )
     email_notifications: Optional[JobsEmailNotifications] = None
-    environments: Optional[List[Jobs]] = Field(
+    environments: Optional[List[JobsEnvironments]] = Field(
         None,
         description='A list of task execution environment specifications that can be referenced by tasks of this job.',
     )

diff --git a/tools/modify_schema.py b/tools/modify_schema.py
@@ -9,7 +9,6 @@ def get_name(path):
     for remove_item in [
         "artifacts",
         "bundle",
-        "environments",
         "include",
         "resources",
         "workspace",