fixing one minor bug :(

weso · Sep 15, 2023 · a5cad56 · a5cad56
1 parent 7ab422f
commit a5cad56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/remote_hdt.rs b/src/remote_hdt.rs
@@ -127,7 +127,7 @@ pub trait Engine {
 
     fn get_second_term(&self, array: ArcArray3, size: usize, term: usize) -> ArcArray1<u8> {
         let mut factor: Array2<u8> = Array2::zeros((size, size));
-        factor[[term, 0]] = 1;
+        factor[[term, term]] = 1; // we place it in the main diagonal
 
         array
             .axis_iter(Axis(0))
@@ -137,7 +137,7 @@ pub trait Engine {
 
     fn get_third_term(&self, array: ArcArray3, size: usize, term: usize) -> ArcArray1<u8> {
         let mut factor: Array2<u8> = Array2::zeros((size, size));
-        factor[[0, term]] = 1;
+        factor[[term, term]] = 1; // we place it in the main diagonal
 
         array
             .axis_iter(Axis(0))