maintain filename

Signed-off-by: Sammy Oina <[email protected]>
ultravioletrs · Aug 8, 2024 · 511f8e6 · 511f8e6
1 parent a8f79a4
commit 511f8e6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 13 deletions.
diff --git a/agent/service.go b/agent/service.go
@@ -194,7 +194,7 @@ func (as *agentService) Data(ctx context.Context, dataset Dataset) error {
 
 	as.computation.Datasets = slices.Delete(as.computation.Datasets, index, index+1)
 
-	f, err := os.Create(fmt.Sprintf("%s/dataset-%d", algorithm.DatasetsDir, index))
+	f, err := os.Create(fmt.Sprintf("%s/%s", algorithm.DatasetsDir, dataset.Filename))
 	if err != nil {
 		return fmt.Errorf("error creating dataset file: %v", err)
 	}

diff --git a/hal/linux/package/agent/agent.mk b/hal/linux/package/agent/agent.mk
@@ -4,8 +4,8 @@
 #
 ################################################################################
 
-AGENT_VERSION = filename
-AGENT_SITE = $(call github,sammyoina,cocos-ai,$(AGENT_VERSION))
+AGENT_VERSION = main
+AGENT_SITE = $(call github,ultravioletrs,cocos,$(AGENT_VERSION))
 
 define AGENT_BUILD_CMDS 	
 	$(MAKE) -C $(@D) agent

diff --git a/test/computations/main.go b/test/computations/main.go
@@ -69,20 +69,11 @@ func (s *svc) Run(ipAdress string, reqChan chan *manager.ServerStreamMessage, au
 			return
 		}
 		dataHash := sha3.Sum256(data)
-		datasets = append(datasets, &manager.Dataset{Hash: dataHash[:], UserKey: pubPem.Bytes, Filename: "iris1.csv"})
+		datasets = append(datasets, &manager.Dataset{Hash: dataHash[:], UserKey: pubPem.Bytes})
 	}
 
 	algoHash := sha3.Sum256(algo)
 
-	// Uncomment this to run tests on the manager service on a SEV enabled backend.
-	reqChan <- &manager.ServerStreamMessage{
-		Message: &manager.ServerStreamMessage_BackendInfoReq{
-			BackendInfoReq: &manager.BackendInfoReq{
-				Id: "1",
-			},
-		},
-	}
-
 	reqChan <- &manager.ServerStreamMessage{
 		Message: &manager.ServerStreamMessage_RunReq{
 			RunReq: &manager.ComputationRunReq{