1 files changed, 10 insertions, 5 deletions
diff --git a/opendc-web/opendc-web-ui/src/data/experiments.js b/opendc-web/opendc-web-ui/src/data/experiments.js
index 4797bacb..a76ea53f 100644
--- a/opendc-web/opendc-web-ui/src/data/experiments.js
+++ b/opendc-web/opendc-web-ui/src/data/experiments.js
@@ -22,21 +22,26 @@
 
 import { useQuery } from 'react-query'
 import { fetchTraces } from '../api/traces'
-import { useAuth } from '../auth'
 import { fetchSchedulers } from '../api/schedulers'
 
 /**
+ * Configure the query defaults for the experiment endpoints.
+ */
+export function configureExperimentClient(queryClient, auth) {
+    queryClient.setQueryDefaults('traces', { queryFn: () => fetchTraces(auth) })
+    queryClient.setQueryDefaults('schedulers', { queryFn: () => fetchSchedulers(auth) })
+}
+
+/**
  * Return the available traces to experiment with.
  */
 export function useTraces() {
-    const auth = useAuth()
-    return useQuery('traces', () => fetchTraces(auth))
+    return useQuery('traces')
 }
 
 /**
  * Return the available schedulers to experiment with.
  */
 export function useSchedulers() {
-    const auth = useAuth()
-    return useQuery('schedulers', () => fetchSchedulers(auth))
+    return useQuery('schedulers')
 }