Skip to content

Commit 82ec577

Browse files
update tests
1 parent db74f2d commit 82ec577

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/models/linear_regression.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
X = rand(100, 5)
77
y = rand(100)
88
model = LinearRegression(X, y)
9-
fit!(model)
9+
fit!(model; save_hyperparameter_distribution_results=false)
1010
β = X \ y
1111
for i in 1:size(X, 2)
1212
@test get_constrained_value(model, "β_$i") .≈ β[i] atol = 1e-5
@@ -27,7 +27,7 @@
2727
10.0
2828
]
2929
model = LinearRegression(X, y)
30-
fit!(model)
30+
fit!(model; save_hyperparameter_distribution_results=false)
3131
forec = forecast(model, [1 6.0])
3232
forec.expected_value[1] == [11.0]
3333
end

test/visualization/forecast.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
air_passengers = CSV.File(StateSpaceModels.AIR_PASSENGERS) |> DataFrame
33
log_air_passengers = log.(air_passengers.passengers)
44
model = BasicStructural(log_air_passengers, 12)
5-
fit!(model)
5+
fit!(model; save_hyperparameter_distribution_results=false)
66
# forecasting
77
forec = forecast(model, 12)
88
rec = RecipesBase.apply_recipe(Dict{Symbol, Any}(), model, forec)

0 commit comments

Comments
 (0)