Update

[ghstack-poisoned]

Author: vmoens

torchrl/services/__init__.py

@@ -20,6 +20,7 @@
     >>> tokenizer = services["tokenizer"]
     >>> result = tokenizer.encode.remote(text)
 """
+from __future__ import annotations
 
 from torchrl.services.base import ServiceBase
 from torchrl.services.ray_service import RayService

torchrl/services/base.py

@@ -2,6 +2,7 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
 
 from abc import ABC, abstractmethod
 from typing import Any

torchrl/services/ray_service.py

@@ -2,6 +2,7 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
 
 from typing import Any