From 0dde5ca6bbac3b1e11c5b04f787d77ad1fc67c74 Mon Sep 17 00:00:00 2001 From: Benjamin-Walker Date: Wed, 24 Sep 2025 11:36:03 +0100 Subject: [PATCH] Fixed switching for linear ncdes --- models/generate_model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/models/generate_model.py b/models/generate_model.py index b6c76b5..02a8f9b 100644 --- a/models/generate_model.py +++ b/models/generate_model.py @@ -102,9 +102,7 @@ def create_model( ), None, ) - elif ( - model_name == "bd_linear_ncde" or "diagonal_linear_ncde" or "dense_linear_ncde" - ): + elif model_name in ["bd_linear_ncde", "diagonal_linear_ncde", "dense_linear_ncde"]: return ( LogLinearCDE( data_dim=data_dim,