Skip to content

Commit

Permalink
Add separate method for post validation
Browse files Browse the repository at this point in the history
Signed-off-by: Partho Sarthi <[email protected]>
  • Loading branch information
parthosa committed Nov 16, 2023
1 parent 33811ec commit e9abdf6
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions user_tools/src/spark_rapids_tools/cmdli/argprocessor.py
Original file line number Diff line number Diff line change
Expand Up @@ -237,11 +237,14 @@ def get_or_set_platform(self) -> CspEnv:
runtime_platform = CspEnv.get_default()
else:
runtime_platform = self.p_args['toolArgs']['platform']
self.post_platform_assignment_validation()
return runtime_platform

# Update argv_cases to reflect the platform and validate post platform assignment
def post_platform_assignment_validation(self):
# Update argv_cases to reflect the platform
self.argv_cases[0] = ArgValueCase.VALUE_A
# Any validation post platform assignment should be done here
self.apply_arg_cases([self.rejected, self.extra])
return runtime_platform


@dataclass
Expand Down

0 comments on commit e9abdf6

Please sign in to comment.