save

hamelsmu · Oct 5, 2023 · 91d33d5 · 91d33d5
1 parent ab2643e
commit 91d33d5
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 34 deletions.
diff --git a/README.md b/README.md
@@ -34,20 +34,6 @@ pip install wandbtocsv
 
     Fetching runs... ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00
     Saved 224 runs to hamelsmu-deepspeed-data-100523.csv
-                                                        id  ... gpu_1_mem
-    0    z0-n_gpu1-gcTrue-seq_len512-bs1-model_sz34-qua...  ...       NaN
-    1    z0-n_gpu1-gcTrue-seq_len512-bs32-model_sz13-qu...  ...       NaN
-    2    z0-n_gpu1-gcTrue-seq_len512-bs8-model_sz13-qua...  ...       NaN
-    3    z0-n_gpu1-gcTrue-seq_len512-bs4-model_sz13-qua...  ...       NaN
-    4    z0-n_gpu1-gcTrue-seq_len512-bs1-model_sz13-qua...  ...       NaN
-    ..                                                 ...  ...       ...
-    219  z0-n_gpu1-gcTrue-seq_len64-bs1-model_sz13-6912...  ...       NaN
-    220  z0-n_gpu1-gcTrue-seq_len64-bs200-model_sz7-327...  ...       NaN
-    221  z0-n_gpu1-gcTrue-seq_len64-bs64-model_sz7-2931...  ...       NaN
-    222  z0-n_gpu1-gcTrue-seq_len64-bs8-model_sz7-45632376  ...       NaN
-    223  z0-n_gpu1-gcTrue-seq_len64-bs1-model_sz7-30283126  ...       NaN
-
-    [224 rows x 200 columns]
 
 ## Bonus, explore your wandb runs in [Datasette](https://datasette.io/)
 

diff --git a/nbs/00_core.ipynb b/nbs/00_core.ipynb
@@ -101,8 +101,7 @@
     "\n",
     "    df = pd.DataFrame(all_runs_data)\n",
     "    df.to_csv(output_file, index=False)\n",
-    "    print(f\"Saved {len(df)} runs to {output_file}\\nPreview:\")\n",
-    "    return df"
+    "    print(f\"Saved {len(df)} runs to {output_file}\")"
    ]
   },
   {

diff --git a/nbs/index.ipynb b/nbs/index.ipynb
@@ -73,22 +73,7 @@
      "output_type": "stream",
      "text": [
       "\u001b[2K\u001b[36mFetching runs...\u001b[0m \u001b[38;2;114;156;31m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[35m100%\u001b[0m \u001b[36m0:00:00\u001b[0m\n",
-      "\u001b[?25hSaved 224 runs to hamelsmu-deepspeed-data-100523.csv\n",
-      "Preview:\n",
-      "                                                    id  ... gpu_2_mem\n",
-      "0    z0-n_gpu1-gcTrue-seq_len512-bs1-model_sz34-qua...  ...       NaN\n",
-      "1    z0-n_gpu1-gcTrue-seq_len512-bs32-model_sz13-qu...  ...       NaN\n",
-      "2    z0-n_gpu1-gcTrue-seq_len512-bs8-model_sz13-qua...  ...       NaN\n",
-      "3    z0-n_gpu1-gcTrue-seq_len512-bs4-model_sz13-qua...  ...       NaN\n",
-      "4    z0-n_gpu1-gcTrue-seq_len512-bs1-model_sz13-qua...  ...       NaN\n",
-      "..                                                 ...  ...       ...\n",
-      "219  z0-n_gpu1-gcTrue-seq_len64-bs1-model_sz13-6912...  ...       NaN\n",
-      "220  z0-n_gpu1-gcTrue-seq_len64-bs200-model_sz7-327...  ...       NaN\n",
-      "221  z0-n_gpu1-gcTrue-seq_len64-bs64-model_sz7-2931...  ...       NaN\n",
-      "222  z0-n_gpu1-gcTrue-seq_len64-bs8-model_sz7-45632376  ...       NaN\n",
-      "223  z0-n_gpu1-gcTrue-seq_len64-bs1-model_sz7-30283126  ...       NaN\n",
-      "\n",
-      "[224 rows x 200 columns]\n"
+      "\u001b[?25hSaved 224 runs to hamelsmu-deepspeed-data-100523.csv\n"
      ]
     }
    ],

diff --git a/wandbtocsv/core.py b/wandbtocsv/core.py
@@ -72,5 +72,4 @@ def fetch_runs(
 
     df = pd.DataFrame(all_runs_data)
     df.to_csv(output_file, index=False)
-    print(f"Saved {len(df)} runs to {output_file}\nPreview:")
-    return df
+    print(f"Saved {len(df)} runs to {output_file}")