#!/bin/bash

FILES_TO_FORMAT="atst/ tests/ app.py"

if [ $1 == "check" ]; then
  pipenv run black --check -q ${FILES_TO_FORMAT}
else
  pipenv run black ${FILES_TO_FORMAT}
fi