Predictor fix

2 files changed, 6 insertions, 7 deletions

diff --git a/backend/api/api/Models/Predictor.cs b/backend/api/api/Models/Predictor.cs
index 4534bd8d..530257b2 100644
--- a/backend/api/api/Models/Predictor.cs
+++ b/backend/api/api/Models/Predictor.cs
@@ -10,8 +10,6 @@ namespace api.Models
 		[BsonRepresentation(BsonType.ObjectId)]//mongo data type to .net
 		public string _id { get; set; }
 		public string uploaderId { get; set; }
-		//public string name { get; set; }
-		//public string description { get; set; }
 		public string[] inputs { get; set; }
 		public string output { get; set; }
 		public bool isPublic { get; set; }
diff --git a/frontend/src/app/_data/Predictor.ts b/frontend/src/app/_data/Predictor.ts
index 55d610ed..c839a635 100644
--- a/frontend/src/app/_data/Predictor.ts
+++ b/frontend/src/app/_data/Predictor.ts
@@ -3,19 +3,20 @@ import { FolderFile } from "./FolderFile";
 export default class Predictor extends FolderFile {
     constructor(
         name: string = 'Novi prediktor',
-        public description: string = '',
+        
+        public uploaderId: string = '',
         public inputs: string[] = [],
         public output: string = '',
         public isPublic: boolean = false,
         public accessibleByLink: boolean = false,
         dateCreated: Date = new Date(),
-        lastUpdated: Date = new Date(),
-        public uploaderId: string = '',
-        //public finalMetrics: Metric[] = []
         public experimentId: string = "",
         public modelId: string = "",
+        public h5FileId: string = "",
+        public metrics: Metric[] = [],
+        public finalMetrics: Metric[] = []
     ) {
-        super(name, dateCreated, lastUpdated);
+        super(name, dateCreated, dateCreated);
     }
 }