Skip to content

Commit

Permalink
Expose 'features' in CLI and environment variable.
Browse files Browse the repository at this point in the history
  • Loading branch information
lohedges committed Jun 24, 2024
1 parent f62a107 commit 7942e5e
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions bin/emle-server
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ try:
except:
port = None
model = os.getenv("EMLE_MODEL")
features = os.getenv("EMLE_FEATURES")
method = os.getenv("EMLE_METHOD")
mm_charges = os.getenv("EMLE_MM_CHARGES")
try:
Expand Down Expand Up @@ -124,6 +125,7 @@ env = {
"host": host,
"port": port,
"model": model,
"features": features,
"method": method,
"mm_charges": mm_charges,
"num_clients": num_clients,
Expand Down Expand Up @@ -177,6 +179,12 @@ parser.add_argument("--port", type=str, help="the port number", required=False)
parser.add_argument(
"--model", type=str, help="path to an EMLE model file", required=False
)
parser.add_argument(
"--features",
type=str,
help="the type of feature used by the model ('soap' or 'aev')",
required=False,
)
parser.add_argument(
"--method",
type=str,
Expand Down

0 comments on commit 7942e5e

Please sign in to comment.