Bugfix

paulsengroup · Sep 24, 2023 · 8c47f11 · 8c47f11
1 parent a138acd
commit 8c47f11
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 10 deletions.
diff --git a/src/libhictk/balancing/include/hictk/balancing/ice.hpp b/src/libhictk/balancing/include/hictk/balancing/ice.hpp
@@ -14,7 +14,7 @@
 namespace hictk::balancing {
 
 class ICE {
-  std::vector<std::size_t> _chrom_offsets{};
+  std::vector<std::uint64_t> _chrom_offsets{};
   std::vector<double> _biases{};
   std::vector<double> _variance{};
   std::vector<double> _scale{};

diff --git a/src/libhictk/balancing/include/hictk/balancing/impl/ice_impl.hpp b/src/libhictk/balancing/include/hictk/balancing/impl/ice_impl.hpp
@@ -145,14 +145,10 @@ inline void ICE::balance_cis(const MatrixT& matrix, const BinTable& bins, std::s
 template <typename File>
 auto ICE::construct_sparse_matrix(const File& f, Type type, std::size_t num_masked_diags)
     -> SparseMatrix {
-  switch (type) {
-    case Type::cis:
-      return construct_sparse_matrix_cis(f, num_masked_diags);
-    case Type::trans:
-      [[fallthrough]];
-    case Type::gw:
-      return construct_sparse_matrix_gw(f, num_masked_diags);
+  if (type == Type::cis) {
+    return construct_sparse_matrix_cis(f, num_masked_diags);
   }
+  return construct_sparse_matrix_gw(f, num_masked_diags);
 }
 
 template <typename File>

diff --git a/src/libhictk/balancing/include/hictk/balancing/sparse_matrix.hpp b/src/libhictk/balancing/include/hictk/balancing/sparse_matrix.hpp
@@ -21,8 +21,8 @@ class SparseMatrix {
   std::vector<double> _counts{};
 
   std::uint32_t _chrom_id{};  // ID of the chromosome that is being procesed
-  std::vector<std::size_t> _chrom_offsets{};
-  std::vector<std::size_t> _bin1_offsets{};
+  std::vector<std::uint64_t> _chrom_offsets{};
+  std::vector<std::uint64_t> _bin1_offsets{};
   mutable std::vector<double> _marg{};
 
   static constexpr auto _gw_id = std::numeric_limits<std::uint32_t>::max();