🐛 minor bug fixes and reformat (#182)

* reformat * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
gradsflow · May 18, 2022 · 60ca633 · 60ca633
1 parent e339b1d
commit 60ca633
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 18 deletions.
diff --git a/examples/nbs/02-TextClassification.ipynb b/examples/nbs/02-TextClassification.ipynb
@@ -49,7 +49,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "5fc8603f",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": [
     "# download_data(\"https://pl-flash-data.s3.amazonaws.com/imdb.zip\", \"./data/\")\n",
@@ -59,6 +63,7 @@
     "    \"sentiment\",\n",
     "    train_file=\"data/imdb/train.csv\",\n",
     "    val_file=\"data/imdb/valid.csv\",\n",
+    "    batch_size=4,\n",
     ")"
    ]
   },

diff --git a/examples/nbs/03-TextSummarization.ipynb b/examples/nbs/03-TextSummarization.ipynb
@@ -3,7 +3,10 @@
   {
    "cell_type": "markdown",
    "metadata": {
-    "id": "xYgiH2TkkX7x"
+    "id": "xYgiH2TkkX7x",
+    "pycharm": {
+     "name": "#%% md\n"
+    }
    },
    "source": [
     "<!--<badge>--><a href=\"https://colab.research.google.com/github/gradsflow/gradsflow/blob/main/examples/nbs/03-TextSummarization.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a><!--</badge>-->\n",
@@ -17,7 +20,10 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
-    "id": "2_posF7Rj8sH"
+    "id": "2_posF7Rj8sH",
+    "pycharm": {
+     "name": "#%%\n"
+    }
    },
    "outputs": [],
    "source": [
@@ -34,14 +40,18 @@
     "    train_file=\"data/xsum/train.csv\",\n",
     "    val_file=\"data/xsum/valid.csv\",\n",
     "    test_file=\"data/xsum/test.csv\",\n",
+    "    batch_size=4,\n",
     ")"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
-    "id": "YCYFfKhDkVVK"
+    "id": "YCYFfKhDkVVK",
+    "pycharm": {
+     "name": "#%%\n"
+    }
    },
    "outputs": [],
    "source": [

diff --git a/examples/nbs/04-RayDataset.ipynb b/examples/nbs/04-RayDataset.ipynb
@@ -3,7 +3,11 @@
   {
    "cell_type": "markdown",
    "id": "9a226830",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
    "source": [
     "<!--<badge>--><a href=\"https://colab.research.google.com/github/gradsflow/gradsflow/blob/main/examples/nbs/04-RayDataset.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a><!--</badge>-->\n",
     "\n",
@@ -40,7 +44,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "1fffc779",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": [
     "import os\n",
@@ -52,14 +60,15 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "1c550a6f",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": [
-    "from torch.utils.data import DataLoader\n",
-    "from gradsflow import data\n",
     "from gradsflow.data.image import (\n",
     "    RayImageFolder,\n",
-    "    image_dataset_from_directory,\n",
     "    get_augmentations,\n",
     ")\n",
     "from torchvision import transforms as T"
@@ -69,7 +78,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "ef8c7353",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [
     {
      "name": "stderr",
@@ -82,7 +95,6 @@
    "source": [
     "transforms = get_augmentations()\n",
     "\n",
-    "\n",
     "ds = RayImageFolder(\n",
     "    \"/Users/aniket/Pictures/data/train/\",\n",
     "    transforms,\n",
@@ -93,7 +105,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "c53a4e2f",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [],
    "source": [
     "iter_ds = iter(ds)"
@@ -103,7 +119,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "31ac27eb",
-   "metadata": {},
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
    "outputs": [
     {
      "data": {

diff --git a/examples/src/tasks/text_classification.py b/examples/src/tasks/text_classification.py
@@ -7,10 +7,7 @@
 
 print("Creating datamodule...")
 datamodule = TextClassificationData.from_csv(
-    "review",
-    "sentiment",
-    train_file="data/imdb/train.csv",
-    val_file="data/imdb/valid.csv",
+    "review", "sentiment", train_file="data/imdb/train.csv", val_file="data/imdb/valid.csv", batch_size=4
 )
 
 suggested_conf = dict(