Update mehtod signature of _files()

Author: soumya-ghosh

mkdocs/docs/api.md

@@ -845,16 +845,8 @@ readable_metrics: [
 [6.0989]]
 ```
 
-<!-- prettier-ignore-start -->
-
 !!! info
-    Content refers to type of content stored by the data file:
-
-    - 0  Data
-    - 1  Position Deletes
-    - 2  Equality Deletes
-
-<!-- prettier-ignore-end -->
+    Content refers to type of content stored by the data file: `0` - `Data`, `1` - `Position Deletes`, `2` - `Equality Deletes`
 
 To show only data files or delete files in the current snapshot, use `table.inspect.data_files()` and `table.inspect.delete_files()` respectively.
 

pyiceberg/table/__init__.py

@@ -4320,9 +4320,7 @@ def history(self) -> "pa.Table":
 
         return pa.Table.from_pylist(history, schema=history_schema)
 
-    def _files(
-        self, snapshot_id: Optional[int] = None, data_file_content_types: Optional[List[DataFileContent]] = None
-    ) -> "pa.Table":
+    def _files(self, snapshot_id: Optional[int] = None, data_file_filter: Optional[Set[DataFileContent]] = None) -> "pa.Table":
         import pyarrow as pa
 
         from pyiceberg.io.pyarrow import schema_to_pyarrow
@@ -4379,7 +4377,7 @@ def _readable_metrics_struct(bound_type: PrimitiveType) -> pa.StructType:
         for manifest_list in snapshot.manifests(io):
             for manifest_entry in manifest_list.fetch_manifest_entry(io):
                 data_file = manifest_entry.data_file
-                if data_file_content_types and data_file.content not in data_file_content_types:
+                if data_file_filter and data_file.content not in data_file_filter:
                     continue
                 column_sizes = data_file.column_sizes or {}
                 value_counts = data_file.value_counts or {}
@@ -4431,10 +4429,10 @@ def files(self, snapshot_id: Optional[int] = None) -> "pa.Table":
         return self._files(snapshot_id)
 
     def data_files(self, snapshot_id: Optional[int] = None) -> "pa.Table":
-        return self._files(snapshot_id, [DataFileContent.DATA])
+        return self._files(snapshot_id, {DataFileContent.DATA})
 
     def delete_files(self, snapshot_id: Optional[int] = None) -> "pa.Table":
-        return self._files(snapshot_id, [DataFileContent.POSITION_DELETES, DataFileContent.EQUALITY_DELETES])
+        return self._files(snapshot_id, {DataFileContent.POSITION_DELETES, DataFileContent.EQUALITY_DELETES})
 
 
 class _ManifestMergeManager(Generic[U]):