From 07f63cc81fd02e8375f0aa95cc285a8fae3b5ce6 Mon Sep 17 00:00:00 2001 From: Nathan Koppel Date: Wed, 28 Jun 2023 18:13:24 -0500 Subject: [PATCH] run cargo fmt --- src/nn/tensor_collection/collection.rs | 2 +- src/nn/tensor_collection/mod.rs | 4 ++-- src/nn/tensor_collection/visitor.rs | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/nn/tensor_collection/collection.rs b/src/nn/tensor_collection/collection.rs index 58df9ee3d..aa08b454d 100644 --- a/src/nn/tensor_collection/collection.rs +++ b/src/nn/tensor_collection/collection.rs @@ -7,7 +7,7 @@ use crate::{ tensor_ops::Device, }; -use super::{ScalarField, ModuleField, ModuleFields, TensorField}; +use super::{ModuleField, ModuleFields, ScalarField, TensorField}; /// A collection of named tensors. Implementing this trait will enable anything /// that operates on tensors, including resetting, counting number of params, updating gradients, diff --git a/src/nn/tensor_collection/mod.rs b/src/nn/tensor_collection/mod.rs index ba4edad4a..b316ea9c4 100644 --- a/src/nn/tensor_collection/mod.rs +++ b/src/nn/tensor_collection/mod.rs @@ -6,8 +6,8 @@ mod collection; mod visitor; mod visitor_impls; -pub use collection::{ScalarOptions, ModuleVisitor, TensorCollection, TensorOptions}; +pub use collection::{ModuleVisitor, ScalarOptions, TensorCollection, TensorOptions}; pub use visitor::{ - ScalarField, ModuleField, ModuleFields, RecursiveWalker, TensorField, TensorViewer, + ModuleField, ModuleFields, RecursiveWalker, ScalarField, TensorField, TensorViewer, TensorVisitor, ViewTensorMut, ViewTensorName, ViewTensorRef, }; diff --git a/src/nn/tensor_collection/visitor.rs b/src/nn/tensor_collection/visitor.rs index 6c85bc55c..466d17ec8 100644 --- a/src/nn/tensor_collection/visitor.rs +++ b/src/nn/tensor_collection/visitor.rs @@ -6,7 +6,7 @@ use crate::{ tensor_ops::Device, }; -use super::{ScalarOptions, ModuleVisitor, TensorCollection, TensorOptions}; +use super::{ModuleVisitor, ScalarOptions, TensorCollection, TensorOptions}; /// A standard [ModuleVisitor] that executes `F` on every [Tensor] encountered. /// `F` must implement [TensorVisitor]