cli: add host option

Author: tharvik

cli/src/args.ts

@@ -11,6 +11,7 @@ interface BenchmarkArguments {
   roundDuration: number
   batchSize: number
   save: boolean
+  host: URL
 }
 
 type BenchmarkUnsafeArguments = Omit<BenchmarkArguments, 'provider'> & {
@@ -28,6 +29,13 @@ const unsafeArgs = parse<BenchmarkUnsafeArguments>(
     roundDuration: { type: Number, alias: 'r', description: 'Round duration (in epochs)', defaultValue: 2 },
     batchSize: { type: Number, alias: 'b', description: 'Training batch size', defaultValue: 10 },
     save: { type: Boolean, alias: 's', description: 'Save logs of benchmark', defaultValue: false },
+    host: {
+      type: (raw: string) => new URL(raw),
+      typeLabel: "URL",
+      description: "Host to connect to",
+      defaultValue: new URL("http://localhost:8080"),
+    },
+
     help: { type: Boolean, optional: true, alias: 'h', description: 'Prints this usage guide' }
   },
   {

cli/src/cli.ts

@@ -48,12 +48,11 @@ async function main<D extends DataType>(
   console.log(`Started ${task.trainingInformation.scheme} training of ${task.id}`)
   console.log({ args })
 
-  const url = new URL('http://localhost:8080')
   const dataSplits = await Promise.all(
     Range(0, numberOfUsers).map(async i => getTaskData(task.id, i))
   )
   const logs = await Promise.all(
-    dataSplits.map(async data => await runUser(task, url, data as Dataset<DataFormat.Raw[D]>))
+    dataSplits.map(async data => await runUser(task, args.host, data as Dataset<DataFormat.Raw[D]>))
   )
 
   if (args.save) {