Add missed dbos_class decorators (#183)

dbos-inc · Jan 22, 2025 · 14c7355 · 14c7355
1 parent a529b4c
commit 14c7355
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/queuedworkflow.py b/tests/queuedworkflow.py
@@ -26,6 +26,7 @@ def default_config() -> ConfigFile:
 q = Queue("testq", concurrency=1, limiter={"limit": 1, "period": 1})
 
 
+@DBOS.dbos_class()
 class WF:
     @staticmethod
     @DBOS.workflow()

diff --git a/tests/test_classdecorators.py b/tests/test_classdecorators.py
@@ -187,6 +187,7 @@ def test_func_admin_r_s(var: str) -> str:
 
 # We can put classes in functions to test decorators for now...
 def test_simple_workflow_static(dbos: DBOS) -> None:
+    @DBOS.dbos_class()
     class DBOSTestClassStatic:
         txn_counter: int = 0
         wf_counter: int = 0