From bba7061d7ff2420f4eb2b6f88362edb71bc6bf25 Mon Sep 17 00:00:00 2001 From: gebele Date: Tue, 18 Jun 2019 15:04:55 +0000 Subject: return training dataset from source --- lib/dataset.rb | 2 +- lib/model.rb | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/dataset.rb b/lib/dataset.rb index 00685b8..51407ca 100644 --- a/lib/dataset.rb +++ b/lib/dataset.rb @@ -16,7 +16,7 @@ get "/api/dataset/:id/?" do halt 400, "Dataset with id: #{params[:id]} not found." unless dataset case @accept when "text/csv", "application/csv" - return dataset.to_csv + return File.read dataset.source else bad_request_error "Mime type #{@accept} is not supported." end diff --git a/lib/model.rb b/lib/model.rb index 9bf4f53..b26447d 100644 --- a/lib/model.rb +++ b/lib/model.rb @@ -20,6 +20,7 @@ end get "/api/model/:id/?" do model = Model::Validation.find params[:id] halt 400, "Model with id: #{params[:id]} not found." unless model + model["training_dataset"] = model.model.training_dataset.id.to_s return model.to_json end -- cgit v1.2.3