Skip to content

Conversation

jpool-nv
Copy link
Contributor

@jpool-nv jpool-nv commented Aug 5, 2025

Removing pickle usage to avoid security issue.

Removing pickle usage to avoid security issue.
Copy link
Collaborator

@Aidyn-A Aidyn-A left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@crcrpar, what do you think?

Comment on lines 67 to 70
cache_dir_path = ASP_CACHE_DIR_DEFAULT
# The user is allowed to set the cache directory via an environment variable.
if environ.get(ASP_CACHE_DIR_ENV_VAR) is not None:
cache_dir_path = environ.get(ASP_CACHE_DIR_ENV_VAR)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Overall looks good to me. I would also recommend to use getenv instead of environ.get to reduce lines.

Suggested change
cache_dir_path = ASP_CACHE_DIR_DEFAULT
# The user is allowed to set the cache directory via an environment variable.
if environ.get(ASP_CACHE_DIR_ENV_VAR) is not None:
cache_dir_path = environ.get(ASP_CACHE_DIR_ENV_VAR)
cache_dir_path = os.getenv(ASP_CACHE_DIR_ENV_VAR, ASP_CACHE_DIR_DEFAULT)

Copy link
Collaborator

@crcrpar crcrpar left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this looks very nice to me

@nWEIdia
Copy link
Collaborator

nWEIdia commented Sep 5, 2025

Merging based on reviewer comments.

@nWEIdia nWEIdia merged commit 214c6b3 into NVIDIA:master Sep 5, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants