Skip to content

Commit c54370c

Browse files
authored
fix: 🐛 Logic Error (#51)
1. Logic Error in DataProto.print_size() Method 2. Missing README_android.md file Co-authored-by: 莫岳恒 <moyueheng@52tt.com>
1 parent a460d3c commit c54370c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565

6666

6767
this_directory = Path(__file__).parent
68-
long_description = (this_directory / "README_android.md").read_text()
68+
long_description = (this_directory / "README.md").read_text()
6969

7070
setup(
7171
name="rl_factory",

verl/protocol.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ def load_from_disk(filepath) -> "DataProto":
290290

291291
def print_size(self, prefix=""):
292292
size_of_tensordict = 0
293-
if self.batch is None:
293+
if self.batch is not None:
294294
for key, tensor in self.batch.items():
295295
size_of_tensordict += tensor.element_size() * tensor.numel()
296296
size_of_numpy_array = 0

0 commit comments

Comments
 (0)