use string when storing lora adapters in simulator (#81)

* use string when storing lora adapters in simulator

Signed-off-by: Maya Barnea <[email protected]>

* fix typo

Signed-off-by: Maya Barnea <[email protected]>

---------

Signed-off-by: Maya Barnea <[email protected]>

Author: mayabar

pkg/llm-d-inference-sim/lora.go

@@ -33,12 +33,13 @@ type unloadLoraRequest struct {
 }
 
 func (s *VllmSimulator) getLoras() []string {
-
 	loras := make([]string, 0)
 
 	s.loraAdaptors.Range(func(key, _ any) bool {
 		if lora, ok := key.(string); ok {
 			loras = append(loras, lora)
+		} else {
+			s.logger.Info("Stored LoRA is not a string", "value", key)
 		}
 		return true
 	})

pkg/llm-d-inference-sim/simulator.go

@@ -190,7 +190,7 @@ func (s *VllmSimulator) parseCommandParamsAndLoadConfig() error {
 	s.config = config
 
 	for _, lora := range config.LoraModules {
-		s.loraAdaptors.Store(lora, "")
+		s.loraAdaptors.Store(lora.Name, "")
 	}
 
 	initRandom(s.config.Seed)