Handle multi-user deployments

kubeflow · May 16, 2024 · 8675b4f · 8675b4f
1 parent a4c69e9
commit 8675b4f
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/frontend/src/pages/RecurringRunList.tsx b/frontend/src/pages/RecurringRunList.tsx
@@ -273,7 +273,19 @@ class RecurringRunList extends React.PureComponent<RecurringRunListProps, Recurr
     let experimentsResponse: V2beta1ListExperimentsResponse;
     let experimentsGetError: string;
     try {
-      experimentsResponse = await Apis.experimentServiceApiV2.listExperiments();
+      if (!this.props.namespaceMask) {
+        // Single-user mode.
+        experimentsResponse = await Apis.experimentServiceApiV2.listExperiments();
+      } else {
+        // Multi-user mode.
+        experimentsResponse = await Apis.experimentServiceApiV2.listExperiments(
+          undefined,
+          undefined,
+          undefined,
+          undefined,
+          this.props.namespaceMask,
+        );
+      }
     } catch (error) {
       experimentsGetError = 'Failed to get associated experiment: ' + (await errorToMessage(error));
     }

diff --git a/frontend/src/pages/RunList.tsx b/frontend/src/pages/RunList.tsx
@@ -409,7 +409,19 @@ class RunList extends React.PureComponent<RunListProps, RunListState> {
     let experimentsResponse: V2beta1ListExperimentsResponse;
     let experimentsGetError: string;
     try {
-      experimentsResponse = await Apis.experimentServiceApiV2.listExperiments();
+      if (!this.props.namespaceMask) {
+        // Single-user mode.
+        experimentsResponse = await Apis.experimentServiceApiV2.listExperiments();
+      } else {
+        // Multi-user mode.
+        experimentsResponse = await Apis.experimentServiceApiV2.listExperiments(
+          undefined,
+          undefined,
+          undefined,
+          undefined,
+          this.props.namespaceMask,
+        );
+      }
     } catch (error) {
       experimentsGetError = 'Failed to get associated experiment: ' + (await errorToMessage(error));
     }