fix: Removed Pipeline Run Version for .63

zenml-io · Jul 31, 2024 · 043c400 · 043c400
1 parent 85ca717
commit 043c400
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 9 deletions.
diff --git a/bundled/tool/zenml_grapher.py b/bundled/tool/zenml_grapher.py
@@ -96,5 +96,4 @@ def to_dict(self) -> GraphResponse:
             "edges": self.edges,
             "status": self.run.body.status,
             "name": self.run.body.pipeline.name,
-            "version": self.run.body.pipeline.body.version,
         }
diff --git a/bundled/tool/zenml_wrappers.py b/bundled/tool/zenml_wrappers.py
@@ -328,7 +328,6 @@ def fetch_pipeline_runs(self, args):
                     "id": str(run.id),
                     "name": run.body.pipeline.name,
                     "status": run.body.status,
-                    "version": run.body.pipeline.body.version,
                     "stackName": run.body.stack.name,
                     "startTime": (
                         run.metadata.start_time.isoformat()
@@ -396,7 +395,6 @@ def get_pipeline_run(self, args: Tuple[str]) -> dict:
                 "id": str(run.id),
                 "name": run.body.pipeline.name,
                 "status": run.body.status,
-                "version": run.body.pipeline.body.version,
                 "stackName": run.body.stack.name,
                 "startTime": (
                     run.metadata.start_time.isoformat()
@@ -434,7 +432,7 @@ def get_pipeline_run_graph(
         """
         try:
             run_id = args[0]
-            run = self.client.get_pipeline_run(run_id, hydrate=True)
+            run = self.client.get_pipeline_run(run_id)
             graph = Grapher(run)
             graph.build_nodes_from_steps()
             graph.build_edges_from_steps()
@@ -485,7 +483,6 @@ def get_run_step(self, args: Tuple[str]) -> Union[RunStepResponse, ErrorResponse
                 "pipeline": {
                     "name": run.body.pipeline.name,
                     "status": run.body.status,
-                    "version": run.body.pipeline.body.version,
                 },
                 "cacheKey": step.metadata.cache_key,
                 "sourceCode": step.metadata.source_code,

diff --git a/src/commands/pipelines/DagRender.ts b/src/commands/pipelines/DagRender.ts
@@ -207,7 +207,7 @@ export default class DagRenderer extends WebviewBase {
     const graph = this.layoutDag(dagData);
     const svg = await this.drawDag(graph);
     const updateButton = dagData.status === 'running' || dagData.status === 'initializing';
-    const title = `${dagData.name} - v${dagData.version}`;
+    const title = `${dagData.name}`;
 
     // And set its HTML content
     panel.webview.html = this.getWebviewContent({

diff --git a/src/types/PipelineTypes.ts b/src/types/PipelineTypes.ts
@@ -25,7 +25,6 @@ export interface PipelineRun {
   id: string;
   name: string;
   status: string;
-  version: string;
   stackName: string;
   startTime: string;
   endTime: string;
@@ -67,7 +66,6 @@ export interface PipelineRunDag {
   edges: Array<DagEdge>;
   status: string;
   name: string;
-  version: string;
 }
 
 export type PipelineRunsResponse = PipelineRunsData | ErrorMessageResponse | VersionMismatchError;
diff --git a/src/views/activityBar/pipelineView/PipelineTreeItems.ts b/src/views/activityBar/pipelineView/PipelineTreeItems.ts
@@ -33,7 +33,7 @@ export class PipelineTreeItem extends vscode.TreeItem {
         : vscode.TreeItemCollapsibleState.Collapsed
     );
     this.tooltip = `${run.name} - Status: ${run.status}`;
-    this.description = `version: ${run.version}, status: ${run.status}`;
+    this.description = `status: ${run.status}`;
     this.iconPath = new vscode.ThemeIcon(PIPELINE_RUN_STATUS_ICONS[run.status]);
     this.children = children;
   }