Skip to content

Commit

Permalink
Fixed formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
uchendui committed Sep 3, 2024
1 parent 0599a69 commit 6b3ee10
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 13 deletions.
14 changes: 8 additions & 6 deletions a2perf/launch/entrypoint.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
import os
import subprocess

from absl import app
from absl import flags
from absl import logging
from absl import app, flags, logging

_ROOT_DIR = flags.DEFINE_string(
"root-dir", None, "Root directory to save metrics and logs from training"
Expand All @@ -22,18 +20,22 @@
_PARTICIPANT_ARGS = flags.DEFINE_string(
"participant-args",
None,
"Additional keyword arguments to pass to the participant's train and inference functions.",
"Additional keyword arguments to pass to the"
" participant's train and inference functions.",
)


def _usage():
return (
"Usage: a2perf <participant_module_path> --root-dir=<root_dir> --submission-gin-config-path=<path> [--participant-args=<k1=v1,k2=v2,...>]\n"
"Usage: a2perf <participant_module_path> "
"--root-dir=<root_dir> "
"--submission-gin-config-path=<path> "
"[--participant-args=<k1=v1,k2=v2,...>]\n"
"\n"
"Options:\n"
"\t--root-dir\tRoot directory for the experiment\n"
"\t--submission-gin-config-path\tPath to the gin configuration file\n"
"\t--participant-args\tAdditional arguments for the participant's train function"
"\t--participant-args\tAdditional arguments for the participant's train function" # noqa: E501
)


Expand Down
7 changes: 0 additions & 7 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,7 @@
"""Sets up the project."""

import importlib.util
import logging
import os
import pathlib
import shutil
import site
import sys
import urllib.request
from distutils.cmd import Command

from setuptools import setup

Expand Down

0 comments on commit 6b3ee10

Please sign in to comment.