dataiku/Dataiku.jl

Description

A Julia package providing functions to interact with Dataiku DSS instances

Overview

This package provides an interface to use DSS remotely and to create recipes/notebooks in Dataiku DSS

Note: This documentation is for remote use. If you want to use the Julia language from within DSS, install the plugin from DSS UI or ask your administrator. More info on the plugin integration here: https://www.dataiku.com/dss/plugins/info/julia.html

Usage

Reading Data

Load a full Dataset

To read a dataset into a DataFrame :

using Dataiku, DataFrames
import Dataiku: get_dataframe

df = get_dataframe(dataset"PROJECTKEY.myDataset")

Keywords parameters

• partitions::AbstractArray : specify the partitions wanted
• infer_types::Bool=true : uses the types detected by CSV.jl rather than the DSS schema
• limit::Integer : Limits the number of rows returned
• ratio::AbstractFloat : Limits the ratio to at n% of the dataset
• sampling::AbstractString="head" : Sampling method, if
  • head returns the first rows of the dataset. Incompatible with ratio parameter.
  • random returns a random sample of the dataset
  • random-column returns a random sample of the dataset. Incompatible with limit parameter.
• truestrings, falsestrings: Vectors of Strings that indicate how true and false values are represented
• sampling_column::AbstractString : Select the column used for "columnwise-random" sampling

Examples :

get_dataframe(dataset"myDataset")
get_dataframe(dataset"myDataset", [:col1, :col2, :col5]; partitions=["2019-02", "2019-03"])
get_dataframe(dataset"PROJECTKEY.myDataset"; infer_types=false, limit=200, sampling="random")

Load a Dataset with Channel

To be able to read data by chunk, without loading all the data in memory. The same keyword parameters can be given.

channel = Dataiku.iter_dataframes(dataset"myDataset", 1000)
first_thousand_row = take!(channel)
second_thousand_row = take!(channel)

It is possible to iterate through the channel

for chunk in channel
    do_stuff(chunk)
end

Iteration row by row with DataFrameRows or tuples is also possible

Dataiku.iter_rows(ds::DSSDataset, columns::AbstractArray=[]; kwargs...)
Dataiku.iter_tuples(ds::DSSDataset, columns::AbstractArray=[]; kwargs...)

Writing Data

Full dataframes

Dataiku.write_with_schema(dataset"myOutputDataset", df)
Dataiku.write_dataframe(dataset"myOutputDataset", df) # does not set the schema of the dataset.

The output dataset must already exist in the project.

Write dataset with Channel

It is also possible to write datasets chunk by chunk to avoid loading full datasets in memory.

input = Dataiku.iter_dataframes(dataset"myInputDataset", 500)
Dataiku.write_dataframe(dataset"myOutputDataset") do chnl
    for chunk in input
        put!(chnl, chunk)
    end
end

Keywords parameters

• partition::AbstractString : specify the partition to write.
• overwrite::Bool=true : if false, appends the data to the already existing dataset.

Project initialization

When using the package inside DSS (recipe or notebook) the projectKey doesn't need to be initialized. Otherwise, you may want to use Dataiku.set_current_project(project"MYPROJECTKEY"). If no project key is initialized, or to use objects for other projects, it's needed to indicate the project during the object creation :

dataset"PROJECTKEY.datasetname"
DSSDataset(project"PROJECTKEY", "datasetname") # equivalent
...

Context initialization

To use the package outside DSS, a url to the instance and an API key or a DKU Ticket is needed. API key can be retrieved in DSS, Administration -> Security

There are 2 ways to initialize it

with config.json file

Create this json file in you're home path $HOME/.dataiku/config.json

{
  "dss_instances": {
    "default": {
      "url": "http://localhost:XXXX/",
      "api_key": "$(APIKEY secret)"
    }
  },
  "default_instance": "default"
}

with init_context function

Dataiku.init_context(url::AbstractString, auth::AbstractString)

Types

The Package implements DSSTypes to interact with different DSS objects

DSSDataset DSSFolder DSSMLTask DSSModelVersion DSSProject DSSSavedModel DSSTrainedModel

These types are only indicators and don't store any data or metadata.

You can have details about what you can do with these types in notebooks or julia REPL like this : ?DSSDataset

For more accessibility, str_macros exist to create most of these types :

• dataset"mydataset" is equivalent to DSSDataset("mydataset")
• dataset"PROJECTKEY.mydataset" => DSSDataset("mydataset", DSSProject("PROJECTKEY"))
• project"PROJECTKEY" => DSSProject("PROJECTKEY")
• folder"XXXXXX" => DSSFolder("XXXXXX")

Tests

A running DSS instance and a config file ($HOME/.dataiku/config.json) are required to run the tests. config.json :

{
  "dss_instances": {
    "default": {
      "url": INSTANCE_URL,
      "api_key": API_KEY_SECRET
    }
  },
  "default_instance": "default"
}