add FAC to column_ner

defog-ai · Nov 1, 2024 · ff37ab1 · ff37ab1
1 parent a34f12b
commit ff37ab1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_utils_pruning.py b/tests/test_utils_pruning.py
@@ -33,6 +33,11 @@ def test_metadata():
             "airport.airport_name,text,name of airport",
             "flight.airport_name,text,name of the airport",
         ],
+        "FAC": [
+            "country.name,text,country name",
+            "airport.airport_name,text,name of airport",
+            "flight.airport_name,text,name of the airport",
+        ],
         "PERSON": ["flight.pilot_name,text,name of the pilot"],
     }
     column_join = {("airport", "country"): [("airport.country_id", "country.id")]}