custom handler

.gitignore

@@ -0,0 +1,17 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+
+# Distribution / packaging
+build/
+dist/
+*.egg-info/
+
+# Environments
+.env
+.venv
+env/
+venv/
+
+# PyCharm
+./.idea/

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/distilbert-base-uncased-emotion-custominf.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="dialignment" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="dialignment" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="dialignment" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/distilbert-base-uncased-emotion-custominf.iml" filepath="$PROJECT_DIR$/.idea/distilbert-base-uncased-emotion-custominf.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

handler.py

@@ -2,29 +2,18 @@ from typing import Dict, List, Any
 from transformers import pipeline
 import holidays
 
-
-class EndpointHandler:
+class EndpointHandler():
     def __init__(self, path=""):
         self.pipeline = pipeline("text-classification", model=path)
         self.holidays = holidays.US()
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        """
-         data args:
-              inputs (:obj: `str`)
-              date (:obj: `str`)
-        Return:
-              A :obj:`list` | `dict`: will be serialized and returned
-        """
-        # get inputs
-        inputs = data.pop("inputs", data)
-        # get additional date field
-        date = data.pop("date", None)
 
-        # check if date exists and if it is a holiday
-        if date is not None and date in self.holidays:
-            return [{"label": "happy", "score": 1}]
+        inputs = data.get("inputs", data)
+        date = data.get("date", None)
 
-        # run normal prediction
+        if date is not None and date in self.holidays:
+            return [{"label":"happy", "score":1}]
+            
         prediction = self.pipeline(inputs)
-        return prediction
+        return prediction

test_handler.py

@@ -0,0 +1,12 @@
+from handler import EndpointHandler
+
+handler = EndpointHandler(".")
+
+non_holiday_payload = {"inputs": "I am so sad", "date": "2022-01-04"}
+holiday_payload = {"inputs": "I am so sad", "date": "2022-07-04"}
+
+non_holiday_response = handler(non_holiday_payload)
+holiday_response = handler(holiday_payload)
+
+print("non_holiday_response:", non_holiday_response)
+print("holiday_response:", holiday_response)