diff --git a/internal/worker/factory.go b/internal/worker/factory.go index 09b7800..22543b0 100644 --- a/internal/worker/factory.go +++ b/internal/worker/factory.go @@ -144,16 +144,16 @@ func NewWorker(cfg *Config, _ string) (*Worker, error) { cancel() return nil, fmt.Errorf("failed to init resource manager: %w", err) } - _ = rm // Resource manager stored for future use worker := &Worker{ - id: cfg.WorkerID, - config: cfg, - logger: logger, - runLoop: runLoop, - metrics: metricsObj, - health: lifecycle.NewHealthMonitor(), - jupyter: jupyterMgr, + id: cfg.WorkerID, + config: cfg, + logger: logger, + runLoop: runLoop, + metrics: metricsObj, + health: lifecycle.NewHealthMonitor(), + resources: rm, + jupyter: jupyterMgr, } // Log GPU configuration diff --git a/internal/worker/worker.go b/internal/worker/worker.go index 6ac6c7b..377e632 100644 --- a/internal/worker/worker.go +++ b/internal/worker/worker.go @@ -9,6 +9,7 @@ import ( "github.com/jfraeys/fetch_ml/internal/jupyter" "github.com/jfraeys/fetch_ml/internal/logging" "github.com/jfraeys/fetch_ml/internal/metrics" + "github.com/jfraeys/fetch_ml/internal/resources" "github.com/jfraeys/fetch_ml/internal/worker/executor" "github.com/jfraeys/fetch_ml/internal/worker/lifecycle" ) @@ -50,6 +51,7 @@ type Worker struct { metrics *metrics.Metrics metricsSrv *http.Server health *lifecycle.HealthMonitor + resources *resources.Manager // Legacy fields for backward compatibility during migration jupyter JupyterManager