Format torch_utils.h

ai4co · May 7, 2022 · b447a61 · b447a61
1 parent d84380e
commit b447a61
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/torchrl/csrc/torch_utils.h b/torchrl/csrc/torch_utils.h
@@ -12,19 +12,23 @@
 namespace torchrl {
 namespace utils {
 
-template <typename T> struct TorchDataType;
+template <typename T>
+struct TorchDataType;
 
-template <> struct TorchDataType<int64_t> {
+template <>
+struct TorchDataType<int64_t> {
   static constexpr torch::ScalarType value = torch::kInt64;
 };
 
-template <> struct TorchDataType<float> {
+template <>
+struct TorchDataType<float> {
   static constexpr torch::ScalarType value = torch::kFloat;
 };
 
-template <> struct TorchDataType<double> {
+template <>
+struct TorchDataType<double> {
   static constexpr torch::ScalarType value = torch::kDouble;
 };
 
-} // namespace utils
-} // namespace torchrl
+}  // namespace utils
+}  // namespace torchrl