Fix test and tsc issue

Author: jpggvilaca

web_ui/src/pages/annotator/providers/prediction-provider/use-inference-server-status.ts

@@ -6,7 +6,7 @@ import { useApplicationServices } from '@geti/core/src/services/application-serv
 import { useQuery, UseQueryResult } from '@tanstack/react-query';
 import { AxiosError } from 'axios';
 
-import { InferenceServerStatusResult } from '../../../../core/annotations/services/prediction-service.interface';
+import { InferenceServerStatusResult } from '../../../../core/annotations/services/inference-service.interface';
 import { ProjectIdentifier } from '../../../../core/projects/core.interface';
 import { useTask } from '../task-provider/task-provider.component';
 

web_ui/src/pages/annotator/providers/selected-media-item-provider/default-selected-media-item-provider.component.tsx

@@ -7,10 +7,8 @@ import QUERY_KEYS from '@geti/core/src/requests/query-keys';
 import { useQuery, UseQueryResult } from '@tanstack/react-query';
 import { noop } from 'lodash-es';
 
-import {
-    ExplanationResult,
-    PredictionResult,
-} from '../../../../core/annotations/services/prediction-service.interface';
+import { ExplanationResult } from '../../../../core/annotations/services/inference-service.interface';
+import { PredictionResult } from '../../../../core/annotations/services/prediction-service.interface';
 import { SelectedMediaItemContext, SelectedMediaItemProps } from './selected-media-item-provider.component';
 import { SelectedMediaItem } from './selected-media-item.interface';
 

web_ui/src/pages/annotator/providers/selected-media-item-provider/use-explanation-query.hook.ts

@@ -36,7 +36,19 @@ export const useExplanationsQuery = ({
         queryFn: async ({ signal }) => {
             if (!mediaItem) throw new Error("Can't fetch undefined media item");
 
-            return await inferenceService.getExplanations(datasetIdentifier, mediaItem, taskId, undefined, signal);
+            try {
+                const explanations = await inferenceService.getExplanations(
+                    datasetIdentifier,
+                    mediaItem,
+                    taskId,
+                    undefined,
+                    signal
+                );
+
+                return explanations ?? [];
+            } catch (_error) {
+                return [];
+            }
         },
         enabled: enabled && !!mediaItem,
         staleTime: 5 * 60_000,

web_ui/src/pages/project-details/components/project-test/test-details-preview/use-test-results-query.hook.test.tsx

@@ -80,7 +80,8 @@ describe('useTestResultsQuery', () => {
         await waitFor(() => {
             expect(mockedInferenceService.getExplanations).toHaveBeenCalled();
             expect(mockedInferenceService.getTestPredictions).toHaveBeenCalled();
-            expect(result.current.predictionsQuery.data).toEqual({ annotations: [mockAnnotation], maps: [] });
+            expect(result.current.predictionsQuery.data).toEqual({ annotations: [mockAnnotation] });
+            expect(result.current.explanationsQuery.data).toEqual([]);
         });
     });
 
@@ -103,7 +104,8 @@ describe('useTestResultsQuery', () => {
         await waitFor(() => {
             expect(mockedInferenceService.getExplanations).toHaveBeenCalled();
             expect(mockedInferenceService.getTestPredictions).toHaveBeenCalled();
-            expect(result.current.predictionsQuery.data).toEqual({ annotations: [], maps: [mockedExplanation] });
+            expect(result.current.predictionsQuery.data).toEqual({ annotations: [] });
+            expect(result.current.explanationsQuery.data).toEqual([mockedExplanation]);
         });
     });
 });

web_ui/src/pages/project-details/components/project-test/test-details-preview/use-test-results-query.hook.ts

@@ -73,14 +73,18 @@ export const useTestResultsQuery = (
     const predictionsQuery = useQuery<PredictionResult, AxiosError>({
         queryKey: QUERY_KEYS.TEST_PREDICTIONS(projectIdentifier, testId, String(testResult?.predictionId)),
         queryFn: async () => {
-            const predictions = await inferenceService.getTestPredictions(
-                datasetIdentifier,
-                labels,
-                testId,
-                String(testResult?.predictionId)
-            );
+            try {
+                const predictions = await inferenceService.getTestPredictions(
+                    datasetIdentifier,
+                    labels,
+                    testId,
+                    String(testResult?.predictionId)
+                );
 
-            return { annotations: predictions };
+                return { annotations: predictions ?? [] };
+            } catch (_error) {
+                return { annotations: [] };
+            }
         },
         enabled: isNonEmptyString(testResult?.predictionId),
     });