Merge pull request #112 from cliveseldon/release_0.1

Version 0.1.0
SeldonIO · May 6, 2021 · 1c02e3f · 1c02e3f
2 parents 377eaa7 + 7d4298f
commit 1c02e3f
Show file tree

Hide file tree

Showing 8 changed files with 109 additions and 988 deletions.
diff --git a/docs/examples/custom-model/README.ipynb b/docs/examples/custom-model/README.ipynb
@@ -115,30 +115,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "sample: 100%|██████████| 3000/3000 [00:09<00:00, 325.84it/s, 3 steps of size 7.77e-01. acc. prob=0.91]\n"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\n",
-      "                mean       std    median      5.0%     95.0%     n_eff     r_hat\n",
-      "         a     -0.00      0.11     -0.00     -0.17      0.17   1794.52      1.00\n",
-      "        bM      0.35      0.13      0.35      0.14      0.56   1748.73      1.00\n",
-      "     sigma      0.94      0.10      0.94      0.77      1.09   2144.79      1.00\n",
-      "\n",
-      "Number of divergences: 0\n"
-     ]
-    }
-   ],
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "import os\n",
     "from tempo.utils import logger\n",
@@ -328,7 +307,7 @@
    "outputs": [],
    "source": [
     "from tempo.serve.loader import save\n",
-    "save(numpyro_divorce, save_env=False)"
+    "save(numpyro_divorce)"
    ]
   },
   {

diff --git a/docs/examples/custom-model/README.md b/docs/examples/custom-model/README.md
@@ -218,7 +218,7 @@ We'll deploy first to Docker to test.
 
 ```python
 from tempo.serve.loader import save
-save(numpyro_divorce, save_env=False)
+save(numpyro_divorce)
 ```
 
 

diff --git a/docs/examples/custom-model/src/tempo.py b/docs/examples/custom-model/src/tempo.py
@@ -49,7 +49,6 @@ def load_numpyro_divorce():
         for k, v in raw_samples.items():
             samples[k] = np.array(v)
 
-        print(model_function.__module__)
         numpyro_divorce.context.predictive_dist = Predictive(model_function, samples)
 
     return numpyro_divorce