Refactor quickstart tutorial notebook (#234)

Co-authored-by: Altay Sansal <[email protected]>
TGSAI · Nov 20, 2024 · 6d00bfc · 6d00bfc
1 parent 115eafc
commit 6d00bfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/tutorials/creation.ipynb b/docs/tutorials/creation.ipynb
@@ -225,8 +225,7 @@
    "mimetype": "text/x-python",
    "name": "python",
    "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.12.3"
+   "pygments_lexer": "ipython3"
   }
  },
  "nbformat": 4,

diff --git a/docs/tutorials/quickstart.ipynb b/docs/tutorials/quickstart.ipynb
@@ -300,6 +300,7 @@
     "    binary_header_fields=[\n",
     "        HeaderField(name=\"sample_interval\", byte=17, format=\"int16\"),\n",
     "        HeaderField(name=\"samples_per_trace\", byte=21, format=\"int16\"),\n",
+    "        HeaderField(name=\"data_sample_format\", byte=25, format=\"int16\"),\n",
     "        HeaderField(name=\"num_extended_text_headers\", byte=305, format=\"int16\"),\n",
     "    ],\n",
     "    trace_header_fields=[\n",
@@ -374,7 +375,8 @@
     "binary_header = sgy.binary_header\n",
     "\n",
     "rng = default_rng()\n",
-    "indices = rng.integers(0, sgy.num_traces, size=5_000).tolist()\n",
+    "indices = rng.choice(sgy.num_traces, size=5_000, replace=False)\n",
+    "indices.sort()\n",
     "\n",
     "traces = sgy.trace[indices]"
    ]