From d595aab5947aa7a8ddfd49ef8b6376a75713de4a Mon Sep 17 00:00:00 2001
From: jrudar <37217429+jrudar@users.noreply.github.com>
Date: Thu, 31 Aug 2023 16:30:44 -0400
Subject: [PATCH] Update TreeOrdination.py

- Corrected code for Issue 'Transformation Bug'
---
 TreeOrdination/TreeOrdination.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/TreeOrdination/TreeOrdination.py b/TreeOrdination/TreeOrdination.py
index 9ebde97..c4bc5d4 100644
--- a/TreeOrdination/TreeOrdination.py
+++ b/TreeOrdination/TreeOrdination.py
@@ -40,7 +40,10 @@ def basic_transform(X, transformer, exclude_col):
         X_no_transform = X_in[:, excl_range]
 
         if X_transformed.ndim == 1:
-            X_transformed = [X_transformed]
+            X_transformed = X_transformed.reshape(-1,1)
+
+        if X_no_transform.ndim == 1:
+             X_no_transform = X_no_transform.rehsape(-1,1)
 
         X_transformed = np.hstack((X_transformed, X_no_transform))