Upload folder using huggingface_hub

physix/systems/registry.py

@@ -22,8 +22,7 @@ SYSTEM_REGISTRY: dict[str, SystemFactory] = {
     "charged_b_field": ChargedInBField,
 }
 
-# Only systems the model was actually trained on (confirmed from W&B run jhsqenj0).
-# TODO: extend training to other systems before exposing them here.
+
 SUPPORTED_SYSTEMS: tuple[str, ...] = (
     "free_fall",
     "simple_pendulum",

train/physix_train_colab.ipynb

@@ -291,6 +291,9 @@
         "    token=os.environ.get(\"HF_TOKEN\"),\n",
         ")\n",
         "\n",
+        "# Verify the bits cell 6 needs are present before continuing.\n",
+        "assert (PHYSIX_LOCAL / \"pyproject.toml\").is_file(), f\"missing pyproject.toml in {PHYSIX_LOCAL}\"\n",
+        "assert (PHYSIX_LOCAL / \"physix\" / \"__init__.py\").is_file(), f\"missing physix/ package in {PHYSIX_LOCAL}\"\n",
         "print(\"physix-live source at\", PHYSIX_LOCAL)\n",
         "!ls {PHYSIX_LOCAL}"
       ],
@@ -301,8 +304,21 @@
       "cell_type": "code",
       "metadata": {},
       "source": [
-        "%%capture\n",
-        "!pip install -q --no-deps -e {PHYSIX_LOCAL}"
+        "import subprocess, sys\n",
+        "result = subprocess.run(\n",
+        "    [sys.executable, \"-m\", \"pip\", \"install\", \"--no-deps\", \"-e\", str(PHYSIX_LOCAL)],\n",
+        "    capture_output=True, text=True, check=False,\n",
+        ")\n",
+        "if result.returncode != 0:\n",
+        "    print(\"pip stdout:\\n\", result.stdout)\n",
+        "    print(\"pip stderr:\\n\", result.stderr)\n",
+        "    raise RuntimeError(\n",
+        "        f\"Editable install of {PHYSIX_LOCAL} failed (rc={result.returncode}). \"\n",
+        "        \"See the pip output above. Common causes: (1) cell 5 didn't actually \"\n",
+        "        \"download the bundle; (2) Colab pip is too old (run `!pip install -q \"\n",
+        "        \"--upgrade pip` and retry); (3) a hatchling build backend hiccup — \"\n",
+        "        \"rerun this cell.\")\n",
+        "print(result.stdout.splitlines()[-1] if result.stdout else \"install ok\")"
       ],
       "execution_count": null,
       "outputs": []
@@ -311,8 +327,12 @@
       "cell_type": "code",
       "metadata": {},
       "source": [
-        "import torch, trl, transformers, datasets, wandb, unsloth, physix\n",
-        "print(f\"torch={torch.__version__}  cuda={torch.cuda.is_available()}  device={torch.cuda.get_device_name(0) if torch.cuda.is_available() else None}\")\n",
+        "import unsloth  # MUST come before trl / transformers so its monkey-patches land first\n",
+        "import torch, trl, transformers, datasets, wandb\n",
+        "import physix\n",
+        "\n",
+        "device = torch.cuda.get_device_name(0) if torch.cuda.is_available() else None\n",
+        "print(f\"torch={torch.__version__}  cuda={torch.cuda.is_available()}  device={device}\")\n",
         "print(f\"unsloth={unsloth.__version__}  trl={trl.__version__}  transformers={transformers.__version__}  datasets={datasets.__version__}\")\n",
         "print(f\"physix loaded from {physix.__file__}\")\n",
         "assert trl.__version__ == \"0.24.0\", f\"trl must be pinned to 0.24.0, got {trl.__version__}\""