docs and dependnecies"

"

convokit/redirection/redirectionDemo.ipynb

@@ -13,9 +13,9 @@
    "id": "c4894fbb",
    "metadata": {},
    "source": [
-    "This notebook demonstrates our redirection framework introduced this paper: **Taking a turn for the better: Conversation redirection throughout the course of mental-health therapy.** In the paper, we define redirection as the extent to which speakers shift the immediate focus of the conversation and applied our measure in the context of long-term messaging therapy. In this demo, we provide an initial exploration into how our redirection framework can be applied in other domains in particular to a publicly available dataset of U.S. Supreme Court oral arguments (Danescu-Niculescu-Mizil et al., 2012; Chang et al., 2020). Although court proceedings differ from therapy in terms of topics, goals, and interaction styles, their relatively unstructured and dynamic nature enables an initial exploration of how such discussions are redirected.\n",
+    "This notebook demonstrates our redirection framework introduced this paper: **Taking a turn for the better: Conversation redirection throughout the course of mental-health therapy.** In the paper, we define redirection as the extent to which speakers shift the immediate focus of the conversation and applied our measure in the context of long-term messaging therapy. In this demo, we provide an initial exploration into how our redirection framework can be applied in other domains in particular, to a publicly available dataset of U.S. Supreme Court oral arguments (Danescu-Niculescu-Mizil et al., 2012; Chang et al., 2020). Although court proceedings differ from therapy in terms of topics, goals, and interaction styles, their relatively unstructured and dynamic nature enables an initial exploration of how such discussions are redirected.\n",
     "\n",
-    "In this setting, we focus on the interactions between justices and lawyers. The power dynamics between these distinct roles reflect the asymmetric relationship between therapists and patients in mental-health domains, where one party generally holds more influence over the conversation."
+    "In this setting, we focus on the interactions between justices and lawyers. The power dynamics between these distinct roles reflect the asymmetric relationship between therapists and patients in mental-health domains, where one party generally holds more influence over the direction of the conversation."
    ]
   },
   {
@@ -33,11 +33,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# !pip install -q convokit\n",
-    "# !pip uninstall convokit\n",
-    "!pip install accelerate peft bitsandbytes transformers trl \n",
-    "!pip install git+https://github.com/vianxnguyen/ConvoKit.git\n",
-    "!pip install numpy==1.24"
+    "!pip install git+https://github.com/vianxnguyen/ConvoKit.git"
    ]
   },
   {

docs/source/analysis.rst

@@ -17,3 +17,4 @@ These are the transformers related to generating some analysis of the Corpus.
    PairedPrediction <pairedprediction.rst>
    Ranker <ranker.rst>
    SpeakerConvoDiversity <speakerConvoDiversity.rst>
+   Redirection <redirection.rst>

docs/source/redirection.rst

@@ -0,0 +1,9 @@
+Redirection
+=======================
+
+Implements redirection as described in this `paper <https://www.cs.cornell.edu/~cristian/Redirection_in_Therapy.html>`_.
+
+Example usage: `redirection in supreme court oral arguments <https://github.com/CornellNLP/ConvoKit/blob/master/examples/speaker-convo-attributes/speaker-convo-diversity-demo.ipynb>`_
+
+.. automodule:: convokit.redirection.redirection
+    :members:

setup.py

@@ -67,7 +67,7 @@
         "peft",
         "bitsandbytes",
         "transformers",
-        "trl",
+        "trl>=0.12.2" "tensorflow>=2.18.0",
     ],
     extras_require={
         "craft": ["torch>=0.12"],