Revert the hacky change for N-d tensors

facebookresearch · Oct 18, 2022 · c509ad4 · c509ad4
1 parent 3dc0323
commit c509ad4
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/beanmachine/ppl/diagnostics/tools/utils/model_serializers.py b/src/beanmachine/ppl/diagnostics/tools/utils/model_serializers.py
@@ -13,10 +13,8 @@
 def serialize_bm(samples: MonteCarloSamples) -> Dict[str, List[List[float]]]:
     """
     Convert Bean Machine models to a JSON serializable object.
-
     Args:
         samples (MonteCarloSamples): Output of a model from Bean Machine.
-
     Returns
         Dict[str, List[List[float]]]: The JSON serializable object for use in the
             diagnostics tools.