fix: bug with new freshness props

raptor-ml · Feb 2, 2023 · d10b880 · d10b880
1 parent 0f869ab
commit d10b880
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/labsdk/raptor/decorators.py b/labsdk/raptor/decorators.py
@@ -473,9 +473,9 @@ def decorator(func):
             spec.labels = options['labels']
 
         if 'freshness' in options:
-            spec.freshness = options['freshness']['target']
-            spec.staleness = options['freshness']['invalid_after']
-            spec.timeout = options['freshness']['latency_sla']
+            spec.freshness = options['freshness']['max_age']
+            spec.staleness = options['freshness']['max_stale']
+            spec.timeout = options['freshness']['timeout']
 
         if spec.freshness is None or spec.staleness is None:
             raise Exception('You must specify freshness')