container.yml

# Container.yml for this `brane_visualize` package
# Basic information about this package
name: brane_visualize
version: 1.0.0
kind: ecu

# These files are part of this package
files: 
- visualize.py

# Defines the file to call whenever a function is called
entrypoint: 
  # 'task' means the script should be run synchronously (i.e., blocking)
  kind: task 
  exec: visualize.py 

# Define the dependencies (as Ubuntu packages)
dependencies:
- python3
- python3-yaml
- python3-pip

# Python packages required for this package
install: 
- pip3 install pandas numpy seaborn matplotlib scikit-learn

# details of functions of the package with input and output details
actions: 
  # Creates the image folder 
  create_img:
    command:
      args:
      - create_img
    input:
    output: 
    - name: status
      type: string

  # Creates the graph for EDA tasks
  visualize_EDA:
    command:
      args:
      - visualize_EDA
    input:
    - name: name
      type: string
    output: 
    - name: status
      type: integer
  
  # Creates the graph for visualization 
  visualize_results:
    command:
      args:
      - visualize_results
    input:
    - name: name
      type: string
    - name: mode
      type: string
    output: 
    - name: status
      type: integer