search

stanfordnlp / pyvene

Stanford NLP Python Library for Understanding and Improving PyTorch Models via Interventions
http://pyvene.ai
Apache License 2.0
559 stars 50 forks source link

[P1] Add intervention util unit tests #100

Closed ZhengPeterWang closed 5 months ago

ZhengPeterWang commented 5 months ago

Description

Add intervention utils unit tests

Testing Done

Testing VanillaIntervention, CollectIntervention, BoundlessRotatedSpaceIntervention

Checklist: