[Minor] Fix prints (pytorch#1257)

ai4co · Jun 12, 2023 · c925a09 · c925a09
1 parent 7642bd1
commit c925a09
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/test/conftest.py b/test/conftest.py
@@ -15,8 +15,10 @@ def pytest_sessionfinish(maxprint=50):
     keys = list(CALL_TIMES.keys())
     if len(keys) > 1:
         maxchar = max(*[len(key) for key in keys])
-    else:
+    elif len(keys) == 1:
         maxchar = len(keys[0])
+    else:
+        return
     for i, (key, item) in enumerate(
         sorted(CALL_TIMES.items(), key=lambda x: x[1], reverse=True)
     ):

diff --git a/torchrl/_utils.py b/torchrl/_utils.py
@@ -1,3 +1,8 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
 import collections
 
 import functools