No public description

PiperOrigin-RevId: 659928831
google-research · Aug 6, 2024 · 1cbdccb · 1cbdccb
1 parent 4a4ed5b
commit 1cbdccb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent_based_epidemic_sim/learning/risk_score_learning.ipynb b/agent_based_epidemic_sim/learning/risk_score_learning.ipynb
@@ -327,7 +327,7 @@
     "\n",
     "def project(params):\n",
     "  \"\"\"Project residual parameters to nonnegative orthant.\"\"\"\n",
-    "  params = np.clip(params, 0, np.infty)\n",
+    "  params = np.clip(params, 0, np.inf)\n",
     "  return params\n",
     "\n",
     "def train(data, labels, weights_init, batch_size=100, num_iters=1000, lr=0.01):\n",