From 606855538d2cba7ba64dc00836ca094f9bf0fb4e Mon Sep 17 00:00:00 2001 From: Jan Sprengel Date: Thu, 15 Jan 2026 17:53:32 +0100 Subject: [PATCH] fix formatting, typo in mofa docs --- muon/_core/tools.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/muon/_core/tools.py b/muon/_core/tools.py index e88f055..27792d3 100644 --- a/muon/_core/tools.py +++ b/muon/_core/tools.py @@ -370,7 +370,7 @@ def mofa( use reduced precision (float32) gpu_mode : optional if to use GPU mode - gpu_mode : optional + gpu_device : optional which GPU device to use svi_mode : optional if to use Stochastic Variational Inference (SVI) @@ -404,8 +404,7 @@ def mofa( path to HDF5 file to store the model expectations : optional which nodes should be used to save expectations for (will save only W and Z by default); - possible expectations names - nclude Y, W, Z, Tau, AlphaZ, AlphaW, ThetaW, ThetaZ + possible expectation names include Y, W, Z, Tau, AlphaZ, AlphaW, ThetaW, ThetaZ save_interrupted : optional if to save partially trained model when the training is interrupted verbose : optional