Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/jax abstract system class #405

Merged
merged 25 commits into from
Feb 25, 2022

Conversation

arnupretorius
Copy link
Collaborator

@arnupretorius arnupretorius commented Feb 10, 2022

What?

Abstract system class for new build system for jax agents.

Why?

Create an interface for systems with functions supporting component-wise builds.

How?

Created a new base abstract class for systems.

Extra

Regarding tests, please note that so far I'm following a different approach than what we had previously been following, which is to locate test files alongside feature files. Two reasons for this: (1) to separate it from our TF2 tests and (2) to experiment whether this might be a more preferred method. But happy to chat more about this.

closes #404

@AsadJeewa
Copy link
Contributor

Really glad that we are finally adding a base class for systems.

Copy link
Contributor

@AsadJeewa AsadJeewa left a comment

Choose a reason for hiding this comment

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

Small comments. Simple class but very useful.

mava/core_jax.py Show resolved Hide resolved
mava/core_jax_test.py Outdated Show resolved Hide resolved
@AsadJeewa AsadJeewa self-requested a review February 22, 2022 08:39
AsadJeewa
AsadJeewa previously approved these changes Feb 22, 2022
Copy link
Contributor

@AsadJeewa AsadJeewa left a comment

Choose a reason for hiding this comment

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

Looks good. Just had one small question

mava/core_jax.py Outdated Show resolved Hide resolved
Copy link
Contributor

@KaleabTessera KaleabTessera left a comment

Choose a reason for hiding this comment

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

Thanks @arnupretorius , I left a comment on the tests.

mava/core_jax_test.py Outdated Show resolved Hide resolved
@AsadJeewa AsadJeewa self-requested a review February 24, 2022 14:57
AsadJeewa
AsadJeewa previously approved these changes Feb 24, 2022
Copy link
Contributor

@AsadJeewa AsadJeewa left a comment

Choose a reason for hiding this comment

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

Approved but have a question on the test

mava/core_jax_test.py Show resolved Hide resolved
Copy link
Collaborator

@RuanJohn RuanJohn left a comment

Choose a reason for hiding this comment

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

Thank you for all your effort @arnupretorius! 💪 🔥

Copy link
Contributor

@KaleabTessera KaleabTessera left a comment

Choose a reason for hiding this comment

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

Thanks @arnupretorius 👍

@arnupretorius arnupretorius merged commit 78756f5 into develop Feb 25, 2022
@arnupretorius arnupretorius deleted the feature/jax-abstract-system-class branch February 25, 2022 07:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request size/L
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[FEATURE] Core abstract system class for Jax based systems
4 participants