Add kagglehub (#1334)

http://b/314191303

Author: rosbo

Dockerfile.tmpl

@@ -363,6 +363,7 @@ RUN pip install mpld3 \
         geoplot \
         eli5 \
         kaggle \
+        kagglehub \
         mock \
         pytest && \
     /tmp/clean-layer.sh

tests/test_kagglehub.py

@@ -0,0 +1,15 @@
+import unittest
+from unittest import mock
+
+import kagglehub
+
+
+class TestKagglehub(unittest.TestCase):
+    def test_login(self):
+        with self.assertLogs('kagglehub', level='INFO') as l:
+            with mock.patch("builtins.input") as mock_input:
+                mock_input.side_effect = ["lastplacelarry", "some-key"]
+                # Disabling credentials validation since network access is disabled in unittest.
+                kagglehub.login(validate_credentials=False)
+
+                self.assertIn("credentials set", l.output[0])