From 2f13ad205a37afa8d3d2331bee895a61a518dce5 Mon Sep 17 00:00:00 2001 From: Yarden Date: Wed, 2 Oct 2024 08:15:35 +0200 Subject: [PATCH] Anon --- LICENSE | 2 +- pyproject.toml | 2 +- safe_opax/common/double_buffer.py | 2 +- safe_opax/common/learner.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/LICENSE b/LICENSE index 56a30542..7c9d40ea 100644 --- a/LICENSE +++ b/LICENSE @@ -1,6 +1,6 @@ MIT License -Copyright (c) 2024 Yarden As +Copyright (c) 2024 Anon Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal diff --git a/pyproject.toml b/pyproject.toml index d02d9a42..ab122444 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -2,7 +2,7 @@ name = "act-safe" version = "0.1.0" description = "" -authors = ["Yarden As"] +authors = ["Anon"] readme = "README.md" [tool.poetry.dependencies] diff --git a/safe_opax/common/double_buffer.py b/safe_opax/common/double_buffer.py index 5d60a1ed..3999dda6 100644 --- a/safe_opax/common/double_buffer.py +++ b/safe_opax/common/double_buffer.py @@ -27,7 +27,7 @@ def double_buffer(ds: Iterable[TrajectoryData]) -> Iterator[TrajectoryData]: Batches of sharded device arrays. """ batch = None - # TODO (yarden): should be sharded better + # TODO (anon): should be sharded better # see equinox's docs for DDP devices = jax.local_devices()[0] for next_batch in ds: diff --git a/safe_opax/common/learner.py b/safe_opax/common/learner.py index 2df14ed7..f75f0e14 100644 --- a/safe_opax/common/learner.py +++ b/safe_opax/common/learner.py @@ -27,7 +27,7 @@ def __init__( def grad_step( self, model: PyTree, grads: PyTree, state: optax.OptState ) -> tuple[PyTree, optax.OptState]: - # TODO (yarden): should retain the policy and use params dtype + # TODO (anon): should retain the policy and use params dtype grads = apply_dtype(grads, jnp.float32) updates, new_opt_state = self.optimizer.update(grads, state) all_ok = all_finite(updates)