[workspace]
authors = [
  "Alex Maldonado <alex@scient.ing>",
]
channels = ["conda-forge", "nvidia", "pytorch"]
name = "pdf2markdown"
platforms = [
  "linux-64",
  { name = "linux-64-cuda-12", platform = "linux-64", cuda = "12" },
  { name = "linux-64-cuda-13", platform = "linux-64", cuda = "13" },
  "osx-64",
  "osx-arm64"
]
version = "26.6.30"

[environments]
datalab = ["datalab"]
marker = ["marker"]
chandra = ["chandra"]

[feature.datalab.dependencies]
python = ">=3.10"

[feature.datalab.pypi-dependencies]
datalab-python-sdk = "*"

[feature.marker]
platforms = ["linux-64-cuda-12"]

[feature.marker.pypi-dependencies]
surya-ocr = "*"
pdftext = "*"
ebooklib = "*"
transformers = "*"
pillow = "*"
pydantic = "*"
pydantic-settings = "*"
python-dotenv = "*"
tqdm = "*"
ftfy = "*"
rapidfuzz = "*"
regex = "*"
markdownify = "*"
click = "*"
markdown2 = "*"
filetype = "*"
google-genai = "*"
anthropic = "*"
pre-commit = "*"
scikit-learn = "*"
mammoth = ">=1.9.0"
openpyxl = ">=3.1.5"
python-pptx = ">=1.0.2"
weasyprint = ">=63.1"
openai = ">=1.65.2"
marker-pdf = ">=1.10.2, <2"

[feature.marker.dependencies]
python = ">=3.10"
pytorch-gpu = ">=2.7.0"
cuda-runtime = "==12.8"
cuda-version = "==12.8"
cuda-cudart-dev = ">=12.8"
cuda-nvcc = ">=12.8"
gcc = ">=14.3.0,<14.4"
glib = ">=2.86.4,<3"

[feature.marker.activation.env]
PYTORCH_CUDA_ALLOC_CONF = "expandable_segments:True"
RECOGNITION_BATCH_SIZE = "2"

[feature.chandra]
platforms = ["linux-64-cuda-13"]

[feature.chandra.pypi-dependencies]
beautifulsoup4 = "*"
click = "*"
filetype = "*"
markdownify = "*"
openai = "*"
pillow = "*"
pydantic = "*"
pydantic-settings = "*"
pypdfium2 = "*"
python-dotenv = "*"
six = "*"
chandra-ocr = { version = ">=0.2.0, <0.3", extras = ["hf"] }

[feature.chandra.dependencies]
python = ">=3.10"
pytorch-gpu = ">=2.8.0"
cuda-runtime = "==13.3.0"
cuda-version = "==13.3.0"
cuda-cudart-dev = ">=13.3.0"
cuda-nvcc = ">=13.3.0"
gcc = ">=14.3.0,<14.4"
glib = ">=2.86.4,<3"