Add an explicit treatment input argument to causaltree/forest #711

	name: build

	on: [push, pull_request]

	jobs:
	build:
	# [macos-latest, macos-latest, windows-latest]
	runs-on: ubuntu-latest
	strategy:
	# You can use PyPy versions in python-version.
	# For example, pypy3.10
	matrix:
	python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]

	steps:
	- uses: actions/checkout@v4
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	allow-prereleases: true
	# You can test your matrix by printing the current Python version
	- name: Display Python version
	run: python --version
	- name: Install dependencies
	run: \|
	sudo apt install graphviz
	pip install --upgrade pip
	pip install --upgrade setuptools
	python -m pip install .[test]
	python setup.py build_ext --inplace
	- name: Test with pytest
	run: pytest -vs tests/ --cov causalml/

Provide feedback