From 247d7f04ec91171004df11bf459dd8a8840b9346 Mon Sep 17 00:00:00 2001
From: Boris Sekachev <40690378+bsekachev@users.noreply.github.com>
Date: Tue, 14 Jan 2020 22:39:35 +0300
Subject: [PATCH] Added ability to match many model labels to one task labels
 (#1051)

* Added ability to match many model labels to one task labels

* Fixed grammar
---
 .../model-runner-modal/model-runner-modal.tsx | 44 ++++++++-----------
 .../containers/file-manager/file-manager.tsx  |  5 ++-
 2 files changed, 22 insertions(+), 27 deletions(-)

diff --git a/cvat-ui/src/components/model-runner-modal/model-runner-modal.tsx b/cvat-ui/src/components/model-runner-modal/model-runner-modal.tsx
index 7d2eacd39d1..2fec246ed19 100644
--- a/cvat-ui/src/components/model-runner-modal/model-runner-modal.tsx
+++ b/cvat-ui/src/components/model-runner-modal/model-runner-modal.tsx
@@ -11,6 +11,7 @@ import {
     Select,
     Tooltip,
     Checkbox,
+    notification,
 } from 'antd';
 
 import { Model } from '../../reducers/interfaces';
@@ -118,6 +119,12 @@ export default class ModelRunnerModalComponent extends React.PureComponent<Props
                 .filter((model) => model.name === selectedModel)[0];
 
             if (!selectedModelInstance.primary) {
+                if (!selectedModelInstance.labels.length) {
+                    notification.warning({
+                        message: 'The selected model does not include any lables',
+                    });
+                }
+
                 let taskLabels: string[] = taskInstance.labels
                     .map((label: any): string => label.name);
                 const [defaultMapping, defaultColors]: StringObject[] = selectedModelInstance.labels
@@ -304,37 +311,24 @@ export default class ModelRunnerModalComponent extends React.PureComponent<Props
         const model = selectedModel && models
             .filter((_model): boolean => _model.name === selectedModel)[0];
 
-        const excludedLabels: {
-            model: string[];
-            task: string[];
-        } = {
-            model: [],
-            task: [],
-        };
-
+        const excludedModelLabels: string[] = Object.keys(mapping);
         const withMapping = model && !model.primary;
-        const tags = withMapping ? Object.keys(mapping)
-            .map((modelLabel: string) => {
-                const taskLabel = mapping[modelLabel];
-                excludedLabels.model.push(modelLabel);
-                excludedLabels.task.push(taskLabel);
-                return this.renderMappingTag(
-                    modelLabel,
-                    mapping[modelLabel],
-                );
-            }) : [];
+        const tags = withMapping ? excludedModelLabels
+            .map((modelLabel: string) => this.renderMappingTag(
+                modelLabel,
+                mapping[modelLabel],
+            )) : [];
 
         const availableModelLabels = model ? model.labels
             .filter(
-                (label: string) => !excludedLabels.model.includes(label),
+                (label: string) => !excludedModelLabels.includes(label),
             ) : [];
-        const availableTaskLabels = taskInstance.labels
-            .map(
-                (label: any) => label.name,
-            ).filter((label: string): boolean => !excludedLabels.task.includes(label));
+        const taskLabels = taskInstance.labels.map(
+            (label: any) => label.name,
+        );
 
         const mappingISAvailable = !!availableModelLabels.length
-            && !!availableTaskLabels.length;
+            && !!taskLabels.length;
 
         return (
             <div className='cvat-run-model-dialog'>
@@ -342,7 +336,7 @@ export default class ModelRunnerModalComponent extends React.PureComponent<Props
                 { withMapping && tags}
                 { withMapping
                     && mappingISAvailable
-                    && this.renderMappingInput(availableModelLabels, availableTaskLabels)
+                    && this.renderMappingInput(availableModelLabels, taskLabels)
                 }
                 { withMapping
                     && (
diff --git a/cvat-ui/src/containers/file-manager/file-manager.tsx b/cvat-ui/src/containers/file-manager/file-manager.tsx
index 4e7a9ca5170..bdfff58eaed 100644
--- a/cvat-ui/src/containers/file-manager/file-manager.tsx
+++ b/cvat-ui/src/containers/file-manager/file-manager.tsx
@@ -26,11 +26,12 @@ interface DispatchToProps {
 function mapStateToProps(state: CombinedState): StateToProps {
     function convert(items: ShareItem[], path?: string): TreeNodeNormal[] {
         return items.map((item): TreeNodeNormal => {
-            const key = `${path}${item.name}/`;
+            const isLeaf = item.type !== 'DIR';
+            const key = `${path}${item.name}${isLeaf ? '' : '/'}`;
             return {
                 key,
+                isLeaf,
                 title: item.name || 'root',
-                isLeaf: item.type !== 'DIR',
                 children: convert(item.children, key),
             };
         });