From 087b5c4dce2de6fd0db17f2d7bf331f500af8d82 Mon Sep 17 00:00:00 2001
From: swakhtar <sheikh-waqas.akhtar@hotmail.com>
Date: Wed, 28 Aug 2024 18:05:24 +0500
Subject: [PATCH] Update experiment1.ipynb

---
 kan/experiments/experiment1.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kan/experiments/experiment1.ipynb b/kan/experiments/experiment1.ipynb
index b453cefe9..be9bbf540 100644
--- a/kan/experiments/experiment1.ipynb
+++ b/kan/experiments/experiment1.ipynb
@@ -206,7 +206,7 @@
     }
    },
    "cell_type": "code",
-   "source": "losses = model.train(dataset=dataset, opt='LBFGS', steps=20, device=device)",
+   "source": "losses = model.fit(dataset=dataset, opt='LBFGS', steps=20)",
    "id": "24cde08da2ea047a",
    "outputs": [
     {