tpcav 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

tpcav/concepts.py

@@ -41,6 +41,8 @@ def _construct_motif_concept_dataloader_from_control(
     motif_mode: str,
     batch_size: int,
     num_workers: int,
+    start_buffer=0,
+    end_buffer=0
 ) -> DataLoader:
     """Mirror the motif-based dataloader logic used in the TCAV script."""
     datasets = []
@@ -51,8 +53,8 @@ def _construct_motif_concept_dataloader_from_control(
             motif=motif,
             motif_mode=motif_mode,
             num_motifs=num_motifs,
-            start_buffer=0,
-            end_buffer=0,
+            start_buffer=start_buffer,
+            end_buffer=end_buffer,
             print_warning=False,
             infinite=False,
         )
@@ -190,7 +192,7 @@ class ConceptBuilder:
                     added.append(concept)
         return added
 
-    def build_motif_concept(self, motifs, concept_name, control_regions=None, motif_mode="pwm"):
+    def build_motif_concept(self, motifs, concept_name, control_regions=None, motif_mode="pwm", start_buffer=0, end_buffer=0):
         if control_regions is None:
             if not self.control_concepts:
                 raise ValueError("Call build_control or pass control_regions first.")
@@ -207,6 +209,8 @@ class ConceptBuilder:
             motif_mode=motif_mode,
             batch_size=self.batch_size,
             num_workers=self.num_workers,
+            start_buffer=start_buffer,
+            end_buffer=end_buffer
         )
         concept = Concept(
             id=self._reserve_id(),

{tpcav-0.2.2.dist-info → tpcav-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tpcav
-Version: 0.2.2
+Version: 0.2.3
 Summary: Testing with PCA projected Concept Activation Vectors
 Author-email: Jianyu Yang <yztxwd@gmail.com>
 License-Expression: MIT AND (Apache-2.0 OR BSD-2-Clause)

{tpcav-0.2.2.dist-info → tpcav-0.2.3.dist-info}/RECORD

@@ -1,12 +1,12 @@
 tpcav/__init__.py,sha256=CpHijSyE1HMy8dlvdSaYrwN9gYMGDEJGDdsneNWnqdA,996
 tpcav/cavs.py,sha256=m4Uhur5LwxdDfSpcznjoKCeqg4-UFjDEicpNglQ_7ss,19377
-tpcav/concepts.py,sha256=_ht4UTu2EVJh52JGnKT3PEgDHk4Q-JCpNuHfFOVmzCw,12884
+tpcav/concepts.py,sha256=rvsSQSybIRTvCPj-AS4Eq7G8Kvh3vlEb6PTWshrXNBs,13044
 tpcav/helper.py,sha256=CcNFJEFG00pujUrthBoMInpIBz1mWIG3y5fztaiHO-c,9917
 tpcav/logging_utils.py,sha256=wug7O_5IjxjhOpQr-aq90qKMEUp1EgcPkrv26d8li6Q,281
 tpcav/tpcav_model.py,sha256=N6YmwqeBR8-QVxkV2uoqhYb3WGGMXo0ND6N8V3dIYug,16573
 tpcav/utils.py,sha256=s2TfC-YoH_xa73WuMqvtpuqzx6g3ne12hE90Yg9hToY,21502
-tpcav-0.2.2.dist-info/licenses/LICENSE,sha256=uC-2s0ObLnQzWFKH5aokHXo6CzxlJgeI0P3bIUCZgfU,1064
-tpcav-0.2.2.dist-info/METADATA,sha256=lnI4YDN09v7vOcrc5a1qKfb_t-XNuKjRb31a1yILioY,3502
-tpcav-0.2.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-tpcav-0.2.2.dist-info/top_level.txt,sha256=I9veSE_WsuFYrXlcfRevqtatDyWWZNsWA3dV0CeBXVg,6
-tpcav-0.2.2.dist-info/RECORD,,
+tpcav-0.2.3.dist-info/licenses/LICENSE,sha256=uC-2s0ObLnQzWFKH5aokHXo6CzxlJgeI0P3bIUCZgfU,1064
+tpcav-0.2.3.dist-info/METADATA,sha256=k99nF8XM2SU-8d_rLVRcR7wN2dsetkjubyYO4DdTm50,3502
+tpcav-0.2.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+tpcav-0.2.3.dist-info/top_level.txt,sha256=I9veSE_WsuFYrXlcfRevqtatDyWWZNsWA3dV0CeBXVg,6
+tpcav-0.2.3.dist-info/RECORD,,