update import paths

zillow · May 16, 2024 · eb6c61d · eb6c61d
1 parent f2a5e7e
commit eb6c61d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/examples/test_model.ipynb b/examples/test_model.ipynb
@@ -27,7 +27,9 @@
     "import pandas as pd\n",
     "\n",
     "from fair_housing_guardrail.utils.helper import load_config, load_dataset, load_tokenizer\n",
-    "from fair_housing_guardrail.utils.fair_housing_classification import FairHousingGuardrailClassification"
+    "from fair_housing_guardrail.utils.fair_housing_classification import (\n",
+    "    FairHousingGuardrailClassification,\n",
+    ")"
    ]
   },
   {

diff --git a/examples/train_model.ipynb b/examples/train_model.ipynb
@@ -31,7 +31,9 @@
     "\n",
     "from fair_housing_guardrail.data.json_dataset import JsonDataset\n",
     "from fair_housing_guardrail.utils.helper import load_config, load_dataset, load_tokenizer\n",
-    "from fair_housing_guardrail.utils.fair_housing_classification import FairHousingGuardrailClassification"
+    "from fair_housing_guardrail.utils.fair_housing_classification import (\n",
+    "    FairHousingGuardrailClassification,\n",
+    ")"
    ]
   },
   {