[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

examples/DVCLive-HuggingFace.ipynb

@@ -106,11 +106,23 @@
    "source": [
     "tokenizer = AutoTokenizer.from_pretrained(\"distilbert-base-cased\")\n",
     "\n",
+    "\n",
     "def tokenize_function(examples):\n",
     "    return tokenizer(examples[\"text\"], padding=\"max_length\", truncation=True)\n",
     "\n",
-    "small_train_dataset = dataset[\"train\"].shuffle(seed=42).select(range(2000)).map(tokenize_function, batched=True)\n",
-    "small_eval_dataset = dataset[\"test\"].shuffle(seed=42).select(range(200)).map(tokenize_function, batched=True)"
+    "\n",
+    "small_train_dataset = (\n",
+    "    dataset[\"train\"]\n",
+    "    .shuffle(seed=42)\n",
+    "    .select(range(2000))\n",
+    "    .map(tokenize_function, batched=True)\n",
+    ")\n",
+    "small_eval_dataset = (\n",
+    "    dataset[\"test\"]\n",
+    "    .shuffle(seed=42)\n",
+    "    .select(range(200))\n",
+    "    .map(tokenize_function, batched=True)\n",
+    ")"
    ]
   },
   {
@@ -138,6 +150,7 @@
     "\n",
     "metric = evaluate.load(\"f1\")\n",
     "\n",
+    "\n",
     "def compute_metrics(eval_pred):\n",
     "    logits, labels = eval_pred\n",
     "    predictions = np.argmax(logits, axis=-1)\n",
@@ -178,7 +191,9 @@
     "from transformers.integrations import DVCLiveCallback\n",
     "from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer\n",
     "\n",
-    "model = AutoModelForSequenceClassification.from_pretrained(\"distilbert-base-cased\", num_labels=2)\n",
+    "model = AutoModelForSequenceClassification.from_pretrained(\n",
+    "    \"distilbert-base-cased\", num_labels=2\n",
+    ")\n",
     "for param in model.base_model.parameters():\n",
     "    param.requires_grad = False\n",
     "\n",
@@ -224,7 +239,6 @@
    "outputs": [],
    "source": [
     "from dvclive import Live\n",
-    "from transformers.integrations import DVCLiveCallback\n",
     "\n",
     "lr = 1e-4\n",
     "\n",
@@ -273,13 +287,13 @@
     "import dvc.api\n",
     "import pandas as pd\n",
     "\n",
-    "columns = [\"Experiment\",  \"epoch\", \"eval.f1\"]\n",
+    "columns = [\"Experiment\", \"epoch\", \"eval.f1\"]\n",
     "\n",
     "df = pd.DataFrame(dvc.api.exp_show(), columns=columns)\n",
     "\n",
     "df.dropna(inplace=True)\n",
     "df.reset_index(drop=True, inplace=True)\n",
-    "df\n"
+    "df"
    ]
   },
   {
@@ -302,7 +316,8 @@
    "outputs": [],
    "source": [
     "from IPython.display import HTML\n",
-    "HTML(filename='./dvc_plots/index.html')"
+    "\n",
+    "HTML(filename=\"./dvc_plots/index.html\")"
    ]
   }
  ],