Move data utilities to separate module

hsm207 · May 17, 2020 · c244912 · c244912
1 parent 3693c6a
commit c244912
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/src/Utils.jl b/src/Utils.jl
@@ -7,6 +7,9 @@ using Gen, RCall, Statistics, DataFrames
 
 import StatsBase
 
+include("data.jl")
+using .Data
+
 export get_data,
  get_posterior_samples,
  get_prediction_log_probs, 
@@ -17,10 +20,6 @@ export get_data,
  VariableSpecification,
  waic
 
-build_url(filename) = "https://raw.githubusercontent.com/rmcelreath/rethinking/master/data/$(filename).csv"
-retrieve_file(url) = HTTP.get(url).body |> IOBuffer |> CSV.read
-get_data(filename) = build_url(filename) |> retrieve_file
-
 struct VariableSpecification
  lower_bound::Float64
  upper_bound::Float64

diff --git a/src/data.jl b/src/data.jl
@@ -0,0 +1,10 @@
+module Data
+using HTTP, CSV
+
+export get_data
+
+build_url(filename) = "https://raw.githubusercontent.com/rmcelreath/rethinking/master/data/$(filename).csv"
+retrieve_file(url) = HTTP.get(url).body |> IOBuffer |> CSV.read
+get_data(filename) = build_url(filename) |> retrieve_file
+
+end