Skip to content

Commit d61b2b5

Browse files
committed
Update _nbeatskan_pkg.py
1 parent 8554c8a commit d61b2b5

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytorch_forecasting/models/nbeats/_nbeatskan_pkg.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ class NBeatsKAN_pkg(_BasePtForecaster):
99
_tags = {
1010
"info:name": "NBeatsKAN",
1111
"info:compute": 1,
12+
"info:pred_type": ["point"],
13+
"info:y_type": ["numeric"],
1214
"authors": ["Sohaib-Ahmed21"],
1315
"capability:exogenous": False,
1416
"capability:multivariate": False,

0 commit comments

Comments
 (0)