lib/prophet/stan_backend.rb in prophet-rb-0.3.0 vs lib/prophet/stan_backend.rb in prophet-rb-0.3.1
- old
+ new
@@ -4,11 +4,11 @@
@model = load_model
@logger = logger
end
def load_model
- model_file = File.expand_path("../../vendor/prophet-#{platform}/bin/prophet", __dir__)
+ model_file = File.expand_path("../../vendor/#{platform}/bin/prophet", __dir__)
raise Error, "Platform not supported yet" unless File.exist?(model_file)
CmdStan::Model.new(exe_file: model_file)
end
def fit(stan_init, stan_data, **kwargs)
@@ -137,18 +137,18 @@
def platform
if Gem.win_platform?
"windows"
elsif RbConfig::CONFIG["host_os"] =~ /darwin/i
if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
- "mac-arm"
+ "arm64-darwin"
else
- "mac"
+ "x86_64-darwin"
end
else
if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
- "linux-arm"
+ "aarch64-linux"
else
- "linux"
+ "x86_64-linux"
end
end
end
end
end