Add skeleton alignment task

Author: Jacob-Chmura

aif_gen/__init__.py

@@ -1,7 +1,7 @@
 """AIF-Gen: A tool for generating synthetic, continual RLHF preference datasets."""
 
-__version__ = "0.1.0"
+__version__ = '0.1.0'
 
 __all__ = [
-    "__version__",
+    '__version__',
 ]

aif_gen/main.py

@@ -0,0 +1,4 @@
+from aif_gen.task.alignment_task import AlignmentTask
+from aif_gen.task.domain import Domain
+from aif_gen.task.objective import Objective
+from aif_gen.task.preference import Preference

aif_gen/task/__init__.py

@@ -0,0 +1,35 @@
+from .domain import Domain
+from .objective import Objective
+from .preference import Preference
+
+
+class AlignmentTask:
+    r"""Encapsulates the specification of an alignment problem.
+
+    Args:
+        domain (Domain): Domain of the alignment task.
+        objective (Objective): Objective of the alignment task.
+        preference (Preference): Preference of the alignment task.
+    """
+
+    def __init__(
+        self, domain: Domain, objective: Objective, preference: Preference
+    ) -> None:
+        self._domain = domain
+        self._objective = objective
+        self._preference = preference
+
+    def __str__(self) -> str:
+        return f'Domain: {self.domain}, Objective: {self.objective}, Preference: {self.preference}'
+
+    @property
+    def domain(self) -> Domain:
+        return self._domain
+
+    @property
+    def objective(self) -> Objective:
+        return self._objective
+
+    @property
+    def preference(self) -> Preference:
+        return self._preference

aif_gen/task/alignment_task.py

@@ -0,0 +1,3 @@
+class Domain:
+    def __str__(self) -> str:
+        return 'Mock Domain'

aif_gen/task/domain.py

@@ -0,0 +1,3 @@
+class Objective:
+    def __str__(self) -> str:
+        return 'Mock Objective'

aif_gen/task/objective.py

@@ -0,0 +1,3 @@
+class Preference:
+    def __str__(self) -> str:
+        return 'Mock Preference'

aif_gen/task/preference.py

@@ -0,0 +1,13 @@
+from aif_gen.task import AlignmentTask, Domain, Objective, Preference
+
+
+def test_init():
+    domain = Domain()
+    objective = Objective()
+    preference = Preference()
+
+    task = AlignmentTask(domain, objective, preference)
+    assert (
+        str(task)
+        == 'Domain: Mock Domain, Objective: Mock Objective, Preference: Mock Preference'
+    )

test/test_alignment_task.py

@@ -0,0 +1,6 @@
+import pytest
+
+
+@pytest.mark.skip(reason='Not implemented')
+def test_init():
+    pass