KnowMat2/environment.yml at master · hasan-sayeed/KnowMat2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
name: KnowMat
channels:
  - defaults
  - conda-forge
  - pytorch
  # - fastai
dependencies:
  - python==3.11
  - pip
  # BASICS
  - numpy==2.2.6
  - scipy==1.16.0
  - pandas==2.3.2
  - tqdm==4.67.1
  - click
  - ipython
  - ipython
  # VISUALIZATION
  - matplotlib
  - ipympl  # interactive matplotlib plots
  - seaborn
  # - altair
  # - yellowbrick
  # ML, STATS & DEEP LEARNING
  # - statsmodels
  # - scikit-learn
  # - sktime
  # - tslearn
  # - xgboost
  # - catboost
  # - lightgbm
  # - pytorch
  # - fastai  # activate fastai channel above!
  # - tensorflow
  # - keras
  # - spacy
  # OTHER TOOLS
  # - optuna
  # - dask
  # - snakeviz
  - pip:
     - -e .  # install git checkout of KnowMat in editable mode
     # add here only pip-packages that are not available in conda/conda-forge! E.g.:
    #  - langchain==0.3.7
    #  - langchain-community==0.3.5
    #  - langchain-huggingface==0.1.2
    #  - pymupdf==1.24.13
    # #  - RapidFuzz==3.10.1
    #  - streamlit==1.40.2
    #  - flask==3.1.0
    #  - ollama==0.4.3
    # #  - spacy==3.8.4
    # #  - fuzzywuzzy==0.18.0
    # #  - python-Levenshtein==0.26.1
    #  - sentence-transformers==3.3.1
     - langchain==0.3.26
     - langchain-community==0.3.27
     - langchain-openai==0.3.27
     - langgraph==0.6.10
     - trustcall==0.0.39

    # OPENAI
     - openai==1.93.2

    # DOCLING (Advanced PDF Parsing)
     - docling==2.51.0

    # PYDANTIC (Schema Validation)
     - pydantic==2.11.7

    # UTILITIES
     - python-dotenv==1.1.1
     - colorlog==6.9.0

    # NLP & EMBEDDINGS
     - sentence-transformers==5.0.0

      # PYTORCH (for embeddings and NLP models)
      # Note: If you need CUDA support, install separately:
      # pip install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu118
     - torch==2.8.0
     - torchvision==0.23.0
     # pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121
     # - icecream
     # - jax
     # - numpyro
     # - funsor
     # - neptune-client
     # - neptune-contrib

  # DEVELOPMENT ONLY PACKAGES (could also be kept in a separate environment file)
  - jupyterlab
  - pytest
  - pytest-cov
  - tox
  - pre_commit
  - nbdime
  - nbstripout
  - sphinx
  - recommonmark