Skip to content

Commit

Permalink
Update to arguments
Browse files Browse the repository at this point in the history
  • Loading branch information
wi-re committed Aug 19, 2024
1 parent b01a700 commit 9a0fac9
Show file tree
Hide file tree
Showing 4 changed files with 16 additions and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[project]
name = "BasisConvolution"
version = "0.1.3"
version = "0.1.4"
authors = [
{ name="Rene Winchenbach", email="[email protected]" },
]
Expand Down
4 changes: 4 additions & 0 deletions src/BasisConvolution/util/arguments.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,4 +71,8 @@
parser.add_argument('--scaleShiftLoss', type = bool, default = argparse.SUPPRESS, action=argparse.BooleanOptionalAction, help='Shifting the loop')
parser.add_argument('--integrationScheme', type = str, default = argparse.SUPPRESS, help='Integration scheme')

parser.add_argument('--shiftCFL', type = float, default = argparse.SUPPRESS, help='Integration scheme')
parser.add_argument('--shiftIters', type = int, default = argparse.SUPPRESS, help='Integration scheme')


parser.add_argument('--exportPath', type = str, default = argparse.SUPPRESS, help='Export path')
8 changes: 8 additions & 0 deletions src/BasisConvolution/util/hyperparameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,8 @@ def defaultHyperParameters():
'outputDecoder': None,
'edgeMLP': None,
'vertexMLP': None,
'shiftCFL': 10,
'shiftIters': 1,
'integrationScheme': 'semiImplicitEuler',
}
return hyperParameterDict
Expand Down Expand Up @@ -135,6 +137,9 @@ def parseArguments(args, hyperParameterDict):
hyperParameterDict['activation'] = args.activation if hasattr(args, 'activation') else hyperParameterDict['activation']
hyperParameterDict['exportPath'] = args.exportPath if hasattr(args, 'exportPath') else hyperParameterDict['exportPath']
hyperParameterDict['integrationScheme'] = args.integrationScheme if hasattr(args, 'integrationScheme') else hyperParameterDict['integrationScheme']
hyperParameterDict['shiftCFL'] = args.shiftCFL if hasattr(args, 'shiftCFL') else hyperParameterDict['shiftCFL']
hyperParameterDict['shiftIters'] = args.shiftIters if hasattr(args, 'shiftIters') else hyperParameterDict['shiftIters']


hyperParameterDict['device'] = args.device if hasattr(args, 'device') else hyperParameterDict['device']
# hyperParameterDict['dtype'] = torch.
Expand Down Expand Up @@ -264,6 +269,9 @@ def parseConfig(config, hyperParameterDict):
parseEntry(cfg, 'shifting', 'shiftLoss', hyperParameterDict, 'shiftLoss')
parseEntry(cfg, 'shifting', 'scaleShiftLoss', hyperParameterDict, 'scaleShiftLoss')
parseEntry(cfg, 'shifting', 'integrationScheme', hyperParameterDict, 'integrationScheme')
parseEntry(cfg, 'shifting', 'shiftIters', hyperParameterDict, 'shiftIters')
parseEntry(cfg, 'shifting', 'shiftCFL', hyperParameterDict, 'shiftCFL')

parseEntry(cfg, 'dataset', 'dataIndex', hyperParameterDict, 'dataIndex')
parseEntry(cfg, 'shifting', 'skipLastShift', hyperParameterDict, 'skipLastShift')
parseEntry(cfg, 'loss', 'dxdtLossScaling', hyperParameterDict, 'dxdtLossScaling')
Expand Down
3 changes: 3 additions & 0 deletions upload.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
rm -rf dist/*
python -m build
python -m twine upload dist/*

0 comments on commit 9a0fac9

Please sign in to comment.