fix some annotations

Author: rok

python/pyarrow-stubs/pyarrow/_dataset_parquet.pyi

@@ -46,7 +46,7 @@ class ParquetFileFormat(FileFormat):
         self,
         read_options: ParquetReadOptions | None = None,
         default_fragment_scan_options: ParquetFragmentScanOptions | None = None,
-        **kwargs,
+        **kwargs: Any,
     ) -> None: ...
     @property
     def read_options(self) -> ParquetReadOptions: ...
@@ -60,7 +60,6 @@ class ParquetFileFormat(FileFormat):
     def make_fragment(
         self,
         file: StrPath | IO | Buffer | BufferReader,
-
         filesystem: SupportedFileSystem | None = None,
         partition_expression: Expression | None = None,
         row_groups: Iterable[int] | None = None,

python/pyarrow/tests/test_fs.py

@@ -1440,13 +1440,13 @@ def test_s3_proxy_options(monkeypatch, pickle_module):
         S3FileSystem(proxy_options=('http', 'localhost', 9090))
     # Missing scheme
     with pytest.raises(KeyError):
-        S3FileSystem(proxy_options={'host': 'localhost', 'port': 9090})
+        S3FileSystem(proxy_options={'host': 'localhost', 'port': 9090})  # type: ignore[missing-typed-dict-key]
     # Missing host
     with pytest.raises(KeyError):
-        S3FileSystem(proxy_options={'scheme': 'https', 'port': 9090})
+        S3FileSystem(proxy_options={'scheme': 'https', 'port': 9090})  # type: ignore[missing-typed-dict-key]
     # Missing port
     with pytest.raises(KeyError):
-        S3FileSystem(proxy_options={'scheme': 'http', 'host': 'localhost'})
+        S3FileSystem(proxy_options={'scheme': 'http', 'host': 'localhost'})  # type: ignore[missing-typed-dict-key]
     # Invalid proxy URI (invalid scheme httpsB)
     with pytest.raises(pa.ArrowInvalid):
         S3FileSystem(proxy_options='httpsB://localhost:9000')

python/pyarrow/tests/test_scalars.py

@@ -424,7 +424,9 @@ def test_timestamp():
         expected = pd.Timestamp('2000-01-01 12:34:56')
 
         assert arrow_arr[0].as_py() == expected
-        assert cast(pa.TimestampScalar, arrow_arr[0]).value * 1000**i == expected.value
+        value = cast(pa.TimestampScalar, arrow_arr[0]).value
+        assert value is not None
+        assert value * 1000**i == expected.value
 
         tz = 'America/New_York'
         arrow_type = pa.timestamp(unit, tz=tz)
@@ -436,7 +438,9 @@ def test_timestamp():
                     .tz_convert(tz))
 
         assert arrow_arr[0].as_py() == expected
-        assert cast(pa.TimestampScalar, arrow_arr[0]).value * 1000**i == expected.value
+        value = cast(pa.TimestampScalar, arrow_arr[0]).value
+        assert value is not None
+        assert value * 1000**i == expected.value
 
 
 @pytest.mark.nopandas
@@ -531,7 +535,7 @@ def test_duration_nanos_nopandas():
 
 
 def test_month_day_nano_interval():
-    triple = pa.MonthDayNano([-3600, 1800, -50])
+    triple = pa.MonthDayNano([-3600, 1800, -50])  # type: ignore[invalid-argument-type]
     arr = pa.array([triple])
     assert isinstance(arr[0].as_py(), pa.MonthDayNano)
     assert arr[0].as_py() == triple

python/pyarrow/tests/test_schema.py

@@ -548,7 +548,7 @@ def test_schema_equals_invalid_type():
 
     for val in [None, 'string', pa.array([1, 2])]:
         with pytest.raises(TypeError):
-            schema.equals(val)
+            schema.equals(val)  # type: ignore[invalid-argument-type]
 
 
 def test_schema_equality_operators():