CITATION.cff and fix F401 warning

CITATION.cff

@@ -0,0 +1,27 @@
+cff-version: 1.2.0
+title: alignment-handbook
+message: >-
+  Robust recipes to align language models with human and AI
+  preferences.
+type: software
+authors:
+  - given-names: Lewis
+    family-names: Tunstall
+  - given-names: Edward
+    family-names: Beeching
+  - given-names: Nathan
+    family-names: Lambert
+  - given-names: Nazneen
+    family-names: Rajani
+  - given-names: Shengyi
+    family-names: Huang
+  - given-names: Kashif
+    family-names: Rasul
+  - given-names: Alexander
+    name-particle: M.
+    family-names: Rush
+  - given-names: Thomas
+    family-names: Wolf
+repository-code: 'https://github.com/huggingface/alignment-handbook'
+license: Apache-2.0
+version: 0.3.0.dev0

src/alignment/__init__.py

@@ -11,3 +11,21 @@
     get_tokenizer,
     is_adapter_model,
 )
+
+
+__all__ = [
+    "DataArguments",
+    "DPOConfig",
+    "H4ArgumentParser",
+    "ModelArguments",
+    "SFTConfig",
+    "apply_chat_template",
+    "get_datasets",
+    "decontaminate_humaneval",
+    "get_checkpoint",
+    "get_kbit_device_map",
+    "get_peft_config",
+    "get_quantization_config",
+    "get_tokenizer",
+    "is_adapter_model",
+]

src/alignment/release.py

@@ -20,12 +20,20 @@
 
 
 REPLACE_PATTERNS = {
-    "init": (re.compile(r'^__version__\s+=\s+"([^"]+)"\s*$', re.MULTILINE), '__version__ = "VERSION"\n'),
-    "setup": (re.compile(r'^(\s*)version\s*=\s*"[^"]+",', re.MULTILINE), r'\1version="VERSION",'),
+    "init": (
+        re.compile(r'^__version__\s+=\s+"([^"]+)"\s*$', re.MULTILINE),
+        '__version__ = "VERSION"\n',
+    ),
+    "setup": (
+        re.compile(r'^(\s*)version\s*=\s*"[^"]+",', re.MULTILINE),
+        r'\1version="VERSION",',
+    ),
+    "citation": (re.compile(r"^version:\s+[^ ]+", re.MULTILINE), "version: VERSION"),
 }
 REPLACE_FILES = {
     "init": "src/alignment/__init__.py",
     "setup": "setup.py",
+    "citation": "CITATION.cff",
 }
 README_FILE = "README.md"
 
@@ -60,7 +68,9 @@ def pre_release_work(patch=False):
     # First let's get the default version: base version if we are in dev, bump minor otherwise.
     default_version = get_version()
     if patch and default_version.is_devrelease:
-        raise ValueError("Can't create a patch version from the dev branch, checkout a released version!")
+        raise ValueError(
+            "Can't create a patch version from the dev branch, checkout a released version!"
+        )
     if default_version.is_devrelease:
         default_version = default_version.base_version
     elif patch:
@@ -95,8 +105,14 @@ def post_release_work():
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--post_release", action="store_true", help="Whether this is pre or post release.")
-    parser.add_argument("--patch", action="store_true", help="Whether or not this is a patch release.")
+    parser.add_argument(
+        "--post_release",
+        action="store_true",
+        help="Whether this is pre or post release.",
+    )
+    parser.add_argument(
+        "--patch", action="store_true", help="Whether or not this is a patch release."
+    )
     args = parser.parse_args()
     if not args.post_release:
         pre_release_work(patch=args.patch)