"""
Contains all of the metadata for non-bonded terms.
"""

_nb_functional_forms = {
    "LJ": {
        "AB": {
            "form": "A/(r ** 12) - B/(r ** 6)",
            "parameters": ["A", "B"],
            "units": {
                "A": "[energy] * [length] ** 12",
                "B": "[energy] * [length] ** 6",
            },
        },
        "epsilon/sigma": {
            "form": "4.0 * epsilon * ((sigma / R) ** 12 - (sigma / R) ** 6)",
            "parameters": ["epsilon", "sigma"],
            "units": {
                "epsilon": "[energy]",
                "sigma": "[length]",
            },
        },
        # Continue this out
        "description": "This is the classic LJ non-bonded"
    },
    "YANNB": {}
}

nb_metadata = {}

# Valid variables used in all two-body terms
nb_metadata["variables"] = {"r": {"units": "[length]", "description": "Distance between the two indexed atoms."}}

# Add store data
nb_metadata["store_name"] = "nb"
nb_metadata["store_indices"] = {
    "atom_type1": "Index of the first atom.",
    "atom_type2": "Index of the second atom.",
    # "term_index": "Index of nb_type stored in the DataLayer.",
}
# nb_metadata["index_columns"] = ["atom1", "atom2"]
nb_metadata["forms"] = _nb_functional_forms
@@ -321,32 +321,95 @@
 %0A            %7D,%0A
+            %22description%22: %22This is the classic LJ non-bonded%22%0A
         %7D,%0A     
@@ -713,89 +713,15 @@
 out%0A
-        %22description%22: %22This is the classic LJ non-bonded%22%0A    %7D,%0A    %22YANNB%22: %7B%7D
+%0A    %7D,
 %0A%7D%0A%0A

move 'description' in nb metadata