mi-crow 1.0.0__tar.gz → 1.0.0.post1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (186) hide show
  1. mi_crow-1.0.0.post1/.env.example +9 -0
  2. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/.github/workflows/tests.yml +1 -1
  3. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/.gitignore +41 -1
  4. mi_crow-1.0.0.post1/.uv-bin/uv +0 -0
  5. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/PKG-INFO +2 -1
  6. mi_crow-1.0.0.post1/coverage.json +1 -0
  7. mi_crow-1.0.0.post1/docs/api/datasets.md +56 -0
  8. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/best-practices.md +8 -4
  9. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/core-concepts.md +10 -4
  10. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/hooks/advanced.md +5 -5
  11. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/hooks/controllers.md +3 -3
  12. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/hooks/detectors.md +6 -6
  13. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/hooks/fundamentals.md +6 -6
  14. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/hooks/index.md +1 -1
  15. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/hooks/registration.md +8 -8
  16. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/installation.md +7 -1
  17. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/quickstart.md +15 -6
  18. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/troubleshooting.md +24 -6
  19. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/workflows/activation-control.md +7 -7
  20. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/workflows/concept-discovery.md +3 -3
  21. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/workflows/concept-manipulation.md +14 -14
  22. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/workflows/saving-activations.md +38 -1
  23. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/pyproject.toml +21 -0
  24. mi_crow-1.0.0.post1/slurm/prepare_datasets.sh +63 -0
  25. mi_crow-1.0.0.post1/slurm/run_baseline_guards_cpu.sh +56 -0
  26. mi_crow-1.0.0.post1/slurm/run_baseline_guards_gpu.sh +57 -0
  27. mi_crow-1.0.0.post1/slurm/run_debug_bielik_padding_cpu.sh +53 -0
  28. mi_crow-1.0.0.post1/slurm/run_debug_llama_chat_template_cpu.sh +49 -0
  29. mi_crow-1.0.0.post1/slurm/run_direct_prompting.sh +105 -0
  30. mi_crow-1.0.0.post1/slurm/run_direct_prompting_cpu.sh +103 -0
  31. mi_crow-1.0.0.post1/slurm/save_activations.sh +73 -0
  32. mi_crow-1.0.0.post1/slurm/test_activation_saving.sh +29 -0
  33. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507262.err +1 -0
  34. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507262.out +3 -0
  35. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507265.err +1 -0
  36. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507265.out +15 -0
  37. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507266.err +67 -0
  38. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507266.out +12 -0
  39. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507271.err +8 -0
  40. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507271.out +13 -0
  41. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507272.err +59 -0
  42. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507272.out +13 -0
  43. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507277.err +20 -0
  44. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507277.out +6 -0
  45. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507283.err +114 -0
  46. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507283.out +6 -0
  47. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507284.err +647 -0
  48. mi_crow-1.0.0.post1/slurm-logs/sae_save_activations-1507284.out +7 -0
  49. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/datasets/base_dataset.py +71 -1
  50. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/datasets/classification_dataset.py +136 -30
  51. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/datasets/text_dataset.py +165 -24
  52. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/controller.py +12 -7
  53. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/implementations/layer_activation_detector.py +30 -34
  54. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/implementations/model_input_detector.py +87 -87
  55. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/implementations/model_output_detector.py +43 -42
  56. mi_crow-1.0.0.post1/src/mi_crow/hooks/utils.py +150 -0
  57. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/activations.py +174 -77
  58. mi_crow-1.0.0.post1/src/mi_crow/language_model/device_manager.py +119 -0
  59. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/inference.py +18 -5
  60. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/initialization.py +10 -6
  61. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/language_model.py +67 -97
  62. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/layers.py +16 -13
  63. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/persistence.py +4 -2
  64. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/utils.py +5 -5
  65. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/concepts/autoencoder_concepts.py +157 -95
  66. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/concepts/concept_dictionary.py +12 -2
  67. mi_crow-1.0.0.post1/src/mi_crow/mechanistic/sae/concepts/text_heap.py +161 -0
  68. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/modules/topk_sae.py +29 -22
  69. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/sae.py +3 -1
  70. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/sae_trainer.py +362 -29
  71. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/store/local_store.py +11 -5
  72. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/store/store.py +34 -1
  73. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow.egg-info/PKG-INFO +2 -1
  74. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow.egg-info/SOURCES.txt +30 -0
  75. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow.egg-info/requires.txt +1 -0
  76. mi_crow-1.0.0.post1/uv.lock +5552 -0
  77. mi_crow-1.0.0/docs/api/datasets.md +0 -6
  78. mi_crow-1.0.0/src/mi_crow/hooks/utils.py +0 -76
  79. mi_crow-1.0.0/uv.lock +0 -5149
  80. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/.github/workflows/docs.yml +0 -0
  81. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/.github/workflows/publish.yml +0 -0
  82. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/.pre-commit-config.yaml +0 -0
  83. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/.python-version +0 -0
  84. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/MANIFEST.in +0 -0
  85. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/README.md +0 -0
  86. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/api/hooks.md +0 -0
  87. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/api/index.md +0 -0
  88. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/api/language_model.md +0 -0
  89. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/api/sae.md +0 -0
  90. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/api/store.md +0 -0
  91. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/api.md +0 -0
  92. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/experiments/index.md +0 -0
  93. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/experiments/slurm-pipeline.md +0 -0
  94. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/experiments/verify-sae-training.md +0 -0
  95. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/examples.md +0 -0
  96. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/index.md +0 -0
  97. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/workflows/index.md +0 -0
  98. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/guide/workflows/training-sae.md +0 -0
  99. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/docs/index.md +0 -0
  100. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/mkdocs.yml +0 -0
  101. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/raport_implementacji.md +0 -0
  102. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/setup.cfg +0 -0
  103. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/404.html +0 -0
  104. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/api/datasets/index.html +0 -0
  105. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/api/hooks/index.html +0 -0
  106. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/api/index.html +0 -0
  107. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/api/language_model/index.html +0 -0
  108. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/api/sae/index.html +0 -0
  109. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/api/store/index.html +0 -0
  110. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/_mkdocstrings.css +0 -0
  111. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/images/favicon.png +0 -0
  112. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/bundle.e71a0d61.min.js +0 -0
  113. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/bundle.e71a0d61.min.js.map +0 -0
  114. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.ar.min.js +0 -0
  115. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.da.min.js +0 -0
  116. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.de.min.js +0 -0
  117. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.du.min.js +0 -0
  118. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.el.min.js +0 -0
  119. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.es.min.js +0 -0
  120. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.fi.min.js +0 -0
  121. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.fr.min.js +0 -0
  122. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.he.min.js +0 -0
  123. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.hi.min.js +0 -0
  124. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.hu.min.js +0 -0
  125. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.hy.min.js +0 -0
  126. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.it.min.js +0 -0
  127. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.ja.min.js +0 -0
  128. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.jp.min.js +0 -0
  129. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.kn.min.js +0 -0
  130. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.ko.min.js +0 -0
  131. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.multi.min.js +0 -0
  132. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.nl.min.js +0 -0
  133. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.no.min.js +0 -0
  134. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.pt.min.js +0 -0
  135. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.ro.min.js +0 -0
  136. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.ru.min.js +0 -0
  137. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.sa.min.js +0 -0
  138. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.stemmer.support.min.js +0 -0
  139. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.sv.min.js +0 -0
  140. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.ta.min.js +0 -0
  141. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.te.min.js +0 -0
  142. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.th.min.js +0 -0
  143. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.tr.min.js +0 -0
  144. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.vi.min.js +0 -0
  145. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/min/lunr.zh.min.js +0 -0
  146. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/tinyseg.js +0 -0
  147. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/lunr/wordcut.js +0 -0
  148. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/workers/search.7a47a382.min.js +0 -0
  149. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/javascripts/workers/search.7a47a382.min.js.map +0 -0
  150. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/stylesheets/main.618322db.min.css +0 -0
  151. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/stylesheets/main.618322db.min.css.map +0 -0
  152. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/stylesheets/palette.ab4e12ef.min.css +0 -0
  153. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/assets/stylesheets/palette.ab4e12ef.min.css.map +0 -0
  154. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/index.html +0 -0
  155. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/objects.inv +0 -0
  156. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/search/search_index.json +0 -0
  157. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/sitemap.xml +0 -0
  158. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/site/sitemap.xml.gz +0 -0
  159. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/__init__.py +0 -0
  160. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/datasets/__init__.py +0 -0
  161. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/datasets/loading_strategy.py +0 -0
  162. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/__init__.py +0 -0
  163. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/detector.py +0 -0
  164. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/hook.py +0 -0
  165. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/implementations/__init__.py +0 -0
  166. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/hooks/implementations/function_controller.py +0 -0
  167. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/__init__.py +0 -0
  168. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/context.py +0 -0
  169. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/contracts.py +0 -0
  170. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/hook_metadata.py +0 -0
  171. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/language_model/tokenizer.py +0 -0
  172. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/__init__.py +0 -0
  173. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/__init__.py +0 -0
  174. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/autoencoder_context.py +0 -0
  175. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/concepts/__init__.py +0 -0
  176. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/concepts/concept_models.py +0 -0
  177. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/concepts/input_tracker.py +0 -0
  178. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/modules/__init__.py +0 -0
  179. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/modules/l1_sae.py +0 -0
  180. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/training/__init__.py +0 -0
  181. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/mechanistic/sae/training/wandb_logger.py +0 -0
  182. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/store/__init__.py +0 -0
  183. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/store/store_dataloader.py +0 -0
  184. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow/utils.py +0 -0
  185. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow.egg-info/dependency_links.txt +0 -0
  186. {mi_crow-1.0.0 → mi_crow-1.0.0.post1}/src/mi_crow.egg-info/top_level.txt +0 -0
@@ -0,0 +1,9 @@
1
+ # Copy this file to .env and fill in your actual values
2
+
3
+ # Wandb API Key for experiment tracking
4
+ # Get your API key from https://wandb.ai/settings
5
+ SERVER_WANDB_API_KEY=your_wandb_api_key_here
6
+
7
+ # Default Wandb project name (used if not specified in training config)
8
+ # Can also be set via WANDB_PROJECT environment variable
9
+ SERVER_WANDB_PROJECT=sae-training
@@ -43,7 +43,7 @@ jobs:
43
43
  run: |
44
44
  pytest \
45
45
  --maxfail=1 -q \
46
- --cov=. \
46
+ --cov=mi_crow \
47
47
  --cov-report=term \
48
48
  --cov-report=xml:coverage.xml \
49
49
  --cov-report=html \
@@ -92,15 +92,29 @@ wandb
92
92
  outputs
93
93
  /store/
94
94
  /examples/store/
95
+ /experiments/store/
95
96
  /playground/store/
96
97
  /tests/unit/store
97
98
  /tests/store
98
99
 
99
- # experiments
100
+ # Github instructions
101
+ .github/instructions/
102
+
103
+ # Experiments
100
104
  .llm_context/
105
+ .local_results/
101
106
  results/
102
107
  # Exclude data store directories but NOT src/mi_crow/store/ (package code)
103
108
  /experiments/*/store/
109
+ /experiments/store/
110
+
111
+ .local_results/
112
+ .local_activations/
113
+ # All csv files from experiments/pl_dataset_curation folder
114
+ /experiments/pl_dataset_curation/*.csv
115
+ /experiments/pl_dataset_curation/*.md
116
+ # Dataset CSV files
117
+ data/
104
118
 
105
119
  # frontend
106
120
  src/frontend/node_modules/
@@ -109,3 +123,29 @@ src/frontend/build/
109
123
  src/frontend/.next/
110
124
 
111
125
  mi_crow_server
126
+ slurm-logs/
127
+
128
+ # SLURM SAE Pipeline output directories
129
+ experiments/slurm_sae_pipeline/logs/
130
+ experiments/slurm_sae_pipeline/hardware_monitoring_output/
131
+ experiments/slurm_sae_pipeline/inference_results/
132
+ experiments/slurm_sae_pipeline/interactive_naming_results/
133
+ experiments/slurm_sae_pipeline/performance_analysis/
134
+ experiments/slurm_sae_pipeline/results/
135
+ experiments/slurm_sae_pipeline/store/
136
+ experiments/slurm_sae_pipeline/test_inference_output/
137
+ experiments/slurm_sae_pipeline/test_steering_output/
138
+ experiments/slurm_sae_pipeline/test_top_texts/
139
+
140
+ # Generated metrics and reports
141
+ experiments/slurm_sae_pipeline/*_metrics.json
142
+ experiments/slurm_sae_pipeline/dictionaries/**/concepts_report.txt
143
+ experiments/verify_sae_training/*.ipynb
144
+
145
+ # Coverage files
146
+ .coverage.*
147
+ .coverage
148
+
149
+ # uv package manager
150
+ .uv-bin/
151
+ .uv-cache/
Binary file
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mi-crow
3
- Version: 1.0.0
3
+ Version: 1.0.0.post1
4
4
  Summary: Engineer Thesis: Explaining and modifying LLM responses using SAE and concepts.
5
5
  Author-email: Hubert Kowalski <your.email@example.com>, Adam Kaniasty <adam.kaniasty@gmail.com>
6
6
  Requires-Python: >=3.10
@@ -19,6 +19,7 @@ Requires-Dist: wandb>=0.22.1
19
19
  Requires-Dist: pytest>=8.4.2
20
20
  Requires-Dist: pytest-xdist>=3.8.0
21
21
  Requires-Dist: seaborn>=0.13.2
22
+ Requires-Dist: numpy<2.0,>=1.20.0
22
23
  Provides-Extra: dev
23
24
  Requires-Dist: pre-commit>=4.3.0; extra == "dev"
24
25
  Requires-Dist: ruff>=0.13.2; extra == "dev"