fix: code snippets (#704)

zenml-io · Nov 29, 2024 · a5d7d88 · a5d7d88
1 parent 13e5368
commit a5d7d88
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/components/dag-visualizer/ArtifactNode.tsx b/src/components/dag-visualizer/ArtifactNode.tsx
@@ -53,7 +53,7 @@ export function ArtifactNode({ data, selected }: NodeProps<ArtifactVersion & { n
 						className="h-4 w-4 shrink-0 rounded-sm hover:bg-primary-100 active:bg-primary-200"
 						code={`from zenml.client import Client
 							
-artifact = Client().get_artifact_version('${data.id}')
+artifact = Client().get_artifact_version("${data.id}")
 loaded_artifact = artifact.load()`}
 						type="artifact"
 					>

diff --git a/src/components/dag-visualizer/StepNode.tsx b/src/components/dag-visualizer/StepNode.tsx
@@ -50,7 +50,7 @@ export function StepNode({ data, selected }: NodeProps<Step>) {
 						<CopyNodeButton
 							className="h-4 w-4 shrink-0 rounded-sm hover:bg-theme-surface-secondary active:bg-neutral-300"
 							code={`from zenml.client import Client
-step = Client().get_run_step(${data.id})
+step = Client().get_run_step("${data.id}")
 config = step.config`}
 							type="step"
 						>