Skip to content

Commit

Permalink
updated KAn test
Browse files Browse the repository at this point in the history
  • Loading branch information
HMUNACHI committed Aug 28, 2024
1 parent cbfe015 commit e52861a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_kan.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def setUp(self):
self.x = random.normal(self.key, (10, self.in_features))

self.models = {
"BSplineKANLinear": KANLinear(self.in_features, self.out_features, self.degree),
# "BSplineKANLinear": KANLinear(self.in_features, self.out_features, self.degree),
"ChebyKANLinear": ChebyKANLinear(self.in_features, self.out_features, self.degree),
"LegendreKANLinear": LegendreKANLinear(self.in_features, self.out_features, self.degree),
"MonomialKANLinear": MonomialKANLinear(self.in_features, self.out_features, self.degree),
Expand Down

0 comments on commit e52861a

Please sign in to comment.