From e53d976fcb17e6e175721438563c20372c1cec2d Mon Sep 17 00:00:00 2001
From: poisonphang <17688291+PoisonPhang@users.noreply.github.com>
Date: Thu, 23 May 2024 03:17:31 -0500
Subject: [PATCH] fix(unionlabs): gate custom_query

---
 lib/unionlabs/src/cosmwasm/wasm/union/custom_query.rs | 6 ------
 lib/unionlabs/src/cosmwasm/wasm/union/mod.rs          | 1 +
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/lib/unionlabs/src/cosmwasm/wasm/union/custom_query.rs b/lib/unionlabs/src/cosmwasm/wasm/union/custom_query.rs
index 635972833b..ab98b917c8 100644
--- a/lib/unionlabs/src/cosmwasm/wasm/union/custom_query.rs
+++ b/lib/unionlabs/src/cosmwasm/wasm/union/custom_query.rs
@@ -1,6 +1,5 @@
 use core::fmt::Debug;
 
-#[cfg(feature = "cosmwasm")]
 use cosmwasm_std::{Binary, Deps, QueryRequest};
 
 use crate::bls::BlsPublicKey;
@@ -30,10 +29,8 @@ pub enum UnionCustomQuery {
     },
 }
 
-#[cfg(feature = "cosmwasm")]
 impl cosmwasm_std::CustomQuery for UnionCustomQuery {}
 
-#[cfg(feature = "cosmwasm")]
 pub fn query_fast_aggregate_verify(
     deps: Deps<UnionCustomQuery>,
     public_keys: Vec<Binary>,
@@ -51,7 +48,6 @@ pub fn query_fast_aggregate_verify(
         .map_err(|e| Error::FastAggregateVerify(e.to_string()))
 }
 
-#[cfg(feature = "cosmwasm")]
 pub fn query_aggregate_public_keys(
     deps: Deps<UnionCustomQuery>,
     public_keys: Vec<BlsPublicKey>,
@@ -72,7 +68,6 @@ pub fn query_aggregate_public_keys(
 }
 
 #[cfg(feature = "stargate")]
-#[cfg(feature = "cosmwasm")]
 use {
     crate::{
         encoding::{Decode, DecodeAs, Proto},
@@ -87,7 +82,6 @@ use {
 
 #[allow(clippy::missing_panics_doc)]
 #[cfg(feature = "stargate")]
-#[cfg(feature = "cosmwasm")]
 pub fn query_consensus_state<T>(
     deps: Deps<UnionCustomQuery>,
     env: &Env,
diff --git a/lib/unionlabs/src/cosmwasm/wasm/union/mod.rs b/lib/unionlabs/src/cosmwasm/wasm/union/mod.rs
index b2e7f4ac32..c01577ced1 100644
--- a/lib/unionlabs/src/cosmwasm/wasm/union/mod.rs
+++ b/lib/unionlabs/src/cosmwasm/wasm/union/mod.rs
@@ -1 +1,2 @@
+#[cfg(feature = "cosmwasm")]
 pub mod custom_query;