From 043c4004acf99e6538a075e899d3fedfc15a57fe Mon Sep 17 00:00:00 2001
From: Christopher Perkins <Christopher.Perkins@null.net>
Date: Wed, 31 Jul 2024 09:37:27 -0400
Subject: [PATCH] fix: Removed Pipeline Run Version for .63

---
 bundled/tool/zenml_grapher.py                           | 1 -
 bundled/tool/zenml_wrappers.py                          | 5 +----
 src/commands/pipelines/DagRender.ts                     | 2 +-
 src/types/PipelineTypes.ts                              | 2 --
 src/views/activityBar/pipelineView/PipelineTreeItems.ts | 2 +-
 5 files changed, 3 insertions(+), 9 deletions(-)

diff --git a/bundled/tool/zenml_grapher.py b/bundled/tool/zenml_grapher.py
index a2bde8af..f11dea18 100644
--- 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,
         }
\ No newline at end of file
diff --git a/bundled/tool/zenml_wrappers.py b/bundled/tool/zenml_wrappers.py
index f33fd99c..055e2037 100644
--- 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
index e06d3eec..87598125 100644
--- 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
index dfddb584..19315ce1 100644
--- 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
index 128c50b8..2b096572 100644
--- 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;
   }