# syntax=docker/dockerfile:experimental

ARG FROM_IMAGE_NAME
FROM $FROM_IMAGE_NAME
ENV DEBIAN_FRONTEND=noninteractive

RUN sed -i -e 's/^APT/# APT/' -e 's/^DPkg/# DPkg/' \
      /etc/apt/apt.conf.d/docker-clean


RUN pip3 install --no-cache-dir \
      black==24.4.2 \
      isort

COPY . /opt/megatron-lm

WORKDIR /opt/megatron-lm