change PostRunHooks to PostRunHook

alphatrion/run/__init__.py

@@ -1,5 +1,5 @@
 """Run package."""
 
-from alphatrion.run.hooks import PostRunHooks
+from alphatrion.run.hooks import PostRunHook
 
-__all__ = ["PostRunHooks"]
+__all__ = ["PostRunHook"]

alphatrion/run/hooks.py

@@ -6,7 +6,7 @@
 from alphatrion.runtime.runtime import global_runtime
 
 
-class PostRunHooks:
+class PostRunHook:
     """Library of built-in post-run hooks."""
 
     @staticmethod
@@ -26,7 +26,7 @@ async def train_model():
                     "num_epochs": 10,
                 }
 
-            run = exp.run(train_model, post_run_hooks=[PostRunHooks.sync_metadata])
+            run = exp.run(train_model, post_run_hooks=[PostRunHook.sync_metadata])
             # After completion, run metadata will contain accuracy, loss, num_epochs
 
         :param run_id: UUID of the run

tests/integration/test_run_hooks.py

@@ -7,7 +7,7 @@
 
 import alphatrion as alpha
 from alphatrion.experiment import CraftExperiment, ExperimentConfig
-from alphatrion.run import PostRunHooks
+from alphatrion.run import PostRunHook
 from alphatrion.runtime.runtime import global_runtime
 
 
@@ -43,7 +43,7 @@ async def train_model():
 
     async with CraftExperiment.start("test_hook_experiment") as exp:
         # Create run with sync_metadata hook
-        run = exp.run(train_model, post_run_hooks=[PostRunHooks.sync_metadata])
+        run = exp.run(train_model, post_run_hooks=[PostRunHook.sync_metadata])
         await exp.wait()
 
         # Verify run completed
@@ -72,7 +72,7 @@ async def task_with_string_result():
 
     async with CraftExperiment.start("test_hook_non_dict") as exp:
         run = exp.run(
-            task_with_string_result, post_run_hooks=[PostRunHooks.sync_metadata]
+            task_with_string_result, post_run_hooks=[PostRunHook.sync_metadata]
         )
         await exp.wait()
 
@@ -98,7 +98,7 @@ async def task2():
         return {"task": "task2", "accuracy": 0.94}
 
     # Configure experiment with sync_metadata hook
-    config = ExperimentConfig(post_run_hooks=[PostRunHooks.sync_metadata])
+    config = ExperimentConfig(post_run_hooks=[PostRunHook.sync_metadata])
 
     async with CraftExperiment.start("test_exp_hooks", config=config) as exp:
         run1 = exp.run(task1)
@@ -143,7 +143,7 @@ async def train_model():
     async with CraftExperiment.start("test_custom_hook") as exp:
         # Use both built-in and custom hooks
         run = exp.run(
-            train_model, post_run_hooks=[PostRunHooks.sync_metadata, add_custom_info]
+            train_model, post_run_hooks=[PostRunHook.sync_metadata, add_custom_info]
         )
         await exp.wait()
 
@@ -172,7 +172,7 @@ async def train_model():
         return {"accuracy": 0.96, "loss": 0.04}
 
     async with CraftExperiment.start("test_merge_metadata") as exp:
-        run = exp.run(train_model, post_run_hooks=[PostRunHooks.sync_metadata])
+        run = exp.run(train_model, post_run_hooks=[PostRunHook.sync_metadata])
 
         # Manually add some metadata before run completes
         metadb = global_runtime().metadb
@@ -209,7 +209,7 @@ async def train_model():
 
     async with CraftExperiment.start("test_hook_failure") as exp:
         run = exp.run(
-            train_model, post_run_hooks=[buggy_hook, PostRunHooks.sync_metadata]
+            train_model, post_run_hooks=[buggy_hook, PostRunHook.sync_metadata]
         )
         await exp.wait()
 

tests/unit/run/test_hooks.py

@@ -5,7 +5,7 @@
 
 import pytest
 
-from alphatrion.run.hooks import PostRunHooks
+from alphatrion.run.hooks import PostRunHook
 from alphatrion.storage.sqlstore import SQLStore
 
 
@@ -51,7 +51,7 @@ def test_sync_metadata_with_dict_result(db):
 
     with patch("alphatrion.run.hooks.global_runtime", return_value=mock_runtime):
         # Call the hook
-        PostRunHooks.sync_metadata(run_id, result)
+        PostRunHook.sync_metadata(run_id, result)
 
         # Verify metadata was updated
         run = db.get_run(run_id)
@@ -92,7 +92,7 @@ def test_sync_metadata_with_non_dict_result(db):
 
     with patch("alphatrion.run.hooks.global_runtime", return_value=mock_runtime):
         # Call the hook
-        PostRunHooks.sync_metadata(run_id, result)
+        PostRunHook.sync_metadata(run_id, result)
 
         # Verify metadata was not updated
         run = db.get_run(run_id)
@@ -136,7 +136,7 @@ def test_sync_metadata_merges_with_existing_metadata(db):
 
     with patch("alphatrion.run.hooks.global_runtime", return_value=mock_runtime):
         # Call the hook
-        PostRunHooks.sync_metadata(run_id, result)
+        PostRunHook.sync_metadata(run_id, result)
 
         # Verify metadata was merged, not replaced
         run = db.get_run(run_id)
@@ -193,7 +193,7 @@ def test_hook_signature():
     """Test that sync_metadata has correct signature"""
     import inspect
 
-    sig = inspect.signature(PostRunHooks.sync_metadata)
+    sig = inspect.signature(PostRunHook.sync_metadata)
     params = list(sig.parameters.keys())
 
     # Should have exactly 2 parameters: run_id and result