fixed formatting

keras-team · Jul 1, 2023 · ade5fb9 · ade5fb9
1 parent 4e1afc9
commit ade5fb9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/guides/ipynb/keras_nlp/getting_started.ipynb b/guides/ipynb/keras_nlp/getting_started.ipynb
@@ -184,7 +184,7 @@
     "\n",
     "# Inspect first review\n",
     "# Format is (review text tensor, label tensor)\n",
-    "print(imdb_train.unbatch().take(1).get_single_element())\n"
+    "print(imdb_train.unbatch().take(1).get_single_element())"
    ]
   },
   {
@@ -385,9 +385,7 @@
     ")\n",
     "\n",
     "classifier = keras_nlp.models.BertClassifier.from_preset(\n",
-    "    \"bert_tiny_en_uncased\",\n",
-    "    preprocessor=None,\n",
-    "    num_classes=2\n",
+    "    \"bert_tiny_en_uncased\", preprocessor=None, num_classes=2\n",
     ")\n",
     "classifier.fit(\n",
     "    imdb_train_cached,\n",

diff --git a/guides/keras_nlp/getting_started.py b/guides/keras_nlp/getting_started.py
@@ -250,9 +250,7 @@
 )
 
 classifier = keras_nlp.models.BertClassifier.from_preset(
-    "bert_tiny_en_uncased",
-    preprocessor=None,
-    num_classes=2
+    "bert_tiny_en_uncased", preprocessor=None, num_classes=2
 )
 classifier.fit(
     imdb_train_cached,