fix: remove labels from inference setup

Author: akgohain

client/src/components/InputSelector.js

@@ -81,34 +81,30 @@ function InputSelector(props) {
             }
           />
         </Form.Item>
-        <Form.Item
-          label={
-            <Space align="center">
-              <span>
-                {type === "training" ? "Input Label" : "Input Label (Optional)"}
-              </span>
-              <InlineHelpChat
-                taskKey={type}
-                label="Input Label"
-                yamlKey="DATASET.LABEL_NAME"
-                value={workflow.inputLabel}
-                projectContext={projectContext}
-                taskContext={taskContext}
-              />
-            </Space>
-          }
-        >
-          <UnifiedFileInput
-            placeholder="Please select or input label path"
-            onChange={handleLabelChange}
-            value={getValue(workflow.inputLabel)}
-            selectionType={
-              type === "training" || type === "inference"
-                ? "fileOrDirectory"
-                : "file"
+        {type === "training" && (
+          <Form.Item
+            label={
+              <Space align="center">
+                <span>Input Label</span>
+                <InlineHelpChat
+                  taskKey={type}
+                  label="Input Label"
+                  yamlKey="DATASET.LABEL_NAME"
+                  value={workflow.inputLabel}
+                  projectContext={projectContext}
+                  taskContext={taskContext}
+                />
+              </Space>
             }
-          />
-        </Form.Item>
+          >
+            <UnifiedFileInput
+              placeholder="Please select or input label path"
+              onChange={handleLabelChange}
+              value={getValue(workflow.inputLabel)}
+              selectionType="fileOrDirectory"
+            />
+          </Form.Item>
+        )}
         {type === "training" ? (
           <Form.Item
             label={

client/src/components/InputSelector.test.js

@@ -0,0 +1,98 @@
+import React from "react";
+import { render, screen } from "@testing-library/react";
+
+import InputSelector from "./InputSelector";
+import { AppContext } from "../contexts/GlobalContext";
+
+jest.mock("antd", () => {
+  const Form = ({ children }) => <form>{children}</form>;
+  Form.Item = ({ label, children, help }) => (
+    <div>
+      {label}
+      {children}
+      {help ? <div>{help}</div> : null}
+    </div>
+  );
+
+  return {
+    Form,
+    Space: ({ children }) => <div>{children}</div>,
+  };
+});
+
+jest.mock("./UnifiedFileInput", () => (props) => (
+  <input
+    aria-label={props.placeholder}
+    data-selection-type={props.selectionType}
+    readOnly
+    value={typeof props.value === "string" ? props.value : ""}
+  />
+));
+
+jest.mock("./InlineHelpChat", () => () => null);
+
+beforeAll(() => {
+  Object.defineProperty(window, "matchMedia", {
+    writable: true,
+    value: jest.fn().mockImplementation((query) => ({
+      matches: false,
+      media: query,
+      onchange: null,
+      addListener: jest.fn(),
+      removeListener: jest.fn(),
+      addEventListener: jest.fn(),
+      removeEventListener: jest.fn(),
+      dispatchEvent: jest.fn(),
+    })),
+  });
+});
+
+function renderWithContext(type) {
+  const contextValue = {
+    trainingState: {
+      inputImage: "",
+      inputLabel: "",
+      outputPath: "",
+      logPath: "",
+      checkpointPath: "",
+      setInputImage: jest.fn(),
+      setInputLabel: jest.fn(),
+      setOutputPath: jest.fn(),
+      setLogPath: jest.fn(),
+      setCheckpointPath: jest.fn(),
+    },
+    inferenceState: {
+      inputImage: "",
+      inputLabel: "/tmp/stale-label.tif",
+      outputPath: "",
+      logPath: "",
+      checkpointPath: "",
+      setInputImage: jest.fn(),
+      setInputLabel: jest.fn(),
+      setOutputPath: jest.fn(),
+      setLogPath: jest.fn(),
+      setCheckpointPath: jest.fn(),
+    },
+  };
+
+  return render(
+    <AppContext.Provider value={contextValue}>
+      <InputSelector type={type} />
+    </AppContext.Provider>,
+  );
+}
+
+describe("InputSelector", () => {
+  it("shows an input label field for training", () => {
+    renderWithContext("training");
+
+    expect(screen.getByText("Input Label")).toBeTruthy();
+  });
+
+  it("does not show an input label field for inference", () => {
+    renderWithContext("inference");
+
+    expect(screen.queryByText("Input Label")).toBeNull();
+    expect(screen.queryByText("Input Label (Optional)")).toBeNull();
+  });
+});

client/src/views/ModelInference.js

@@ -54,7 +54,7 @@ function ModelInference({ isInferring, setIsInferring }) {
       applyInputPaths(yamlData, {
         mode: "inference",
         inputImagePath: getPath(inference.inputImage),
-        inputLabelPath: getPath(inference.inputLabel),
+        inputLabelPath: "",
         inputPath: "",
         outputPath: getPath(inference.outputPath),
       });