diff --git a/Examples/README.md b/Examples/README.md index 204fda9..f8c71c1 100644 --- a/Examples/README.md +++ b/Examples/README.md @@ -4,7 +4,8 @@ Runnable samples that complement [docs/api-guide](../docs/api-guide/). | Directory | Purpose | |-----------|---------| -| [IrisDemo](IrisDemo/) | Full SwiftUI macOS app: bundled Python sklearn service, Charts, JSON boundary | +| [Studio](Studio/) | **Flagship app.** Native SwiftUI macOS app on the canonical V2 `pool.` surface — races four scikit-learn models in parallel across Python workers, with live charts, worker activity, and an embedded "behind the curtain" console (+ `studio-cli`). Ships as a notarized DMG with a self-contained Python ([SHIPPING.md](Studio/SHIPPING.md)). | +| [IrisDemo](IrisDemo/) | Minimal SwiftUI macOS app: bundled Python sklearn service via the raw runtime + JSON boundary (the simplest "hello world"; predates the `pool.` surface) | | [CoreRuntimeSmoke](CoreRuntimeSmoke/) | Minimal CLI: `Python.run` + `sys` / `json` | | [ProcessPoolSmoke](ProcessPoolSmoke/) | Minimal CLI: `withProcessPool`, `invokeResult`, `math.sqrt` | | [BridgingRing](BridgingRing/) | Showcase: callbacks + **`WorkerCallbackContext` reentrant** Python↔Swift + **`evalEvents`** generator/progress streams over the pool IPC | diff --git a/Examples/Studio/Package.swift b/Examples/Studio/Package.swift new file mode 100644 index 0000000..a72d706 --- /dev/null +++ b/Examples/Studio/Package.swift @@ -0,0 +1,104 @@ +// swift-tools-version: 6.0 +import PackageDescription +import Foundation + +// SwiftPython Studio — a flagship demo app built on the commercial +// SwiftPythonRuntime xcframework. It showcases the canonical V2 `pool.` +// surface (vendored as `SwiftPythonKit`) by racing several scikit-learn models +// across Python worker processes, live, from Swift. + +private func parseSemVer(_ raw: String) -> Version? { + let core = raw.split(separator: "-", maxSplits: 1).first.map(String.init) ?? raw + let parts = core.split(separator: ".", omittingEmptySubsequences: false) + guard parts.count == 3, let a = Int(parts[0]), let b = Int(parts[1]), let c = Int(parts[2]) else { return nil } + return Version(a, b, c) +} + +private let swiftPythonPackageName = "swiftpython-commercial" + +private struct SwiftPythonDependencyConfig { + let dependency: Package.Dependency + let packageName: String +} + +private func packageNameFromDependencyURL(_ rawURL: String) -> String { + var trimmed = rawURL.trimmingCharacters(in: .whitespacesAndNewlines) + while trimmed.hasSuffix("/") { trimmed.removeLast() } + let base = (trimmed as NSString).lastPathComponent + return base.hasSuffix(".git") ? String(base.dropLast(4)) : base +} + +private func swiftPythonConfig() -> SwiftPythonDependencyConfig { + let env = ProcessInfo.processInfo.environment + if let url = env["SWIFTPYTHON_COMMERCIAL_PACKAGE_URL"]?.trimmingCharacters(in: .whitespacesAndNewlines), + !url.isEmpty, + let versionRaw = env["SWIFTPYTHON_COMMERCIAL_PACKAGE_VERSION"]?.trimmingCharacters(in: .whitespacesAndNewlines), + !versionRaw.isEmpty { + guard let version = parseSemVer(versionRaw) else { + fatalError("SWIFTPYTHON_COMMERCIAL_PACKAGE_VERSION must be MAJOR.MINOR.PATCH") + } + return SwiftPythonDependencyConfig( + dependency: .package(url: url, exact: version), + packageName: packageNameFromDependencyURL(url) + ) + } + return SwiftPythonDependencyConfig( + dependency: .package(name: swiftPythonPackageName, path: "../.."), + packageName: swiftPythonPackageName + ) +} + +private func pythonLibraryDirectory() -> String { + let env = ProcessInfo.processInfo.environment + if let explicit = env["SWIFTPYTHON_PYTHON_LIB_DIR"]?.trimmingCharacters(in: .whitespacesAndNewlines), !explicit.isEmpty { + return explicit + } + if let home = (env["PYTHON_HOME"] ?? env["PYTHONHOME"])?.trimmingCharacters(in: .whitespacesAndNewlines), !home.isEmpty { + let frameworkLib = "\(home)/Frameworks/Python.framework/Versions/3.13/lib" + return FileManager.default.fileExists(atPath: frameworkLib) ? frameworkLib : "\(home)/lib" + } + let candidates = [ + "/opt/homebrew/opt/python@3.13/Frameworks/Python.framework/Versions/3.13/lib", + "/usr/local/opt/python@3.13/Frameworks/Python.framework/Versions/3.13/lib", + "/opt/homebrew/opt/python@3.13/lib", + "/usr/local/opt/python@3.13/lib", + ] + return candidates.first { FileManager.default.fileExists(atPath: $0) } ?? candidates[0] +} + +private let pythonLinkerSettings: [LinkerSetting] = [ + .unsafeFlags(["-L\(pythonLibraryDirectory())", "-lpython3.13"]) +] + +private let config = swiftPythonConfig() + +let package = Package( + name: "Studio", + platforms: [.macOS(.v15)], + dependencies: [config.dependency], + targets: [ + // Vendored canonical V2 pool bindings (generated by `swift-python-gen --surface pool`). + .target( + name: "SwiftPythonKit", + dependencies: [.product(name: "SwiftPythonRuntime", package: config.packageName)] + ), + .executableTarget( + name: "Studio", + dependencies: [ + "SwiftPythonKit", + .product(name: "SwiftPythonRuntime", package: config.packageName), + ], + path: "Sources/Studio", + linkerSettings: pythonLinkerSettings + ), + .executableTarget( + name: "studio-cli", + dependencies: [ + "SwiftPythonKit", + .product(name: "SwiftPythonRuntime", package: config.packageName), + ], + path: "Sources/studio-cli", + linkerSettings: pythonLinkerSettings + ), + ] +) diff --git a/Examples/Studio/README.md b/Examples/Studio/README.md new file mode 100644 index 0000000..ebebaa6 --- /dev/null +++ b/Examples/Studio/README.md @@ -0,0 +1,54 @@ +# SwiftPython Studio + +The flagship demo for the SwiftPython runtime: **Swift driving scikit-learn across +real Python worker processes, live, in a native macOS app.** + +Studio races four classifiers (Random Forest, Gradient Boosting, Logistic +Regression, K-Neighbors) in parallel — each pinned to its own Python worker — +through the canonical V2 `pool.` surface, and shows the accuracy bars, +confusion matrix, worker activity, and the raw pool calls streaming in an embedded +"behind the curtain" console. + +```swift +let model = try await pool.sklearn.ensemble.RandomForestClassifier(n_estimators: 300) +let accuracy = try await model.fit(X: xTrain, y: yTrain).score(X: xTest, y: yTest) +``` + +## What it shows off + +- **`pool.` canonical surface** — `pool.sklearn.ensemble.…`, owned remote + handles, typed results, method chaining (`fit(…).score(…)`), all `.handle`-free. +- **True multi-process parallelism** — wall time ≈ the slowest model, not the sum. +- **The commercial xcframework** — built exactly the way a customer consumes it. + +## Run it (dev) + +Requires Homebrew `python@3.13` with `numpy pandas scikit-learn scipy`. + +```bash +cd Examples/Studio + +# GUI +./scripts/build_app.sh --open + +# Terminal demo (same engine, designed output) +SWIFTPYTHON_WORKER_PATH="$(cd ../.. && pwd)/SwiftPythonWorker" \ + swift run studio-cli +``` + +## Layout + +| Path | Role | +|------|------| +| `Sources/Studio/` | SwiftUI app (bake-off screen + embedded console) | +| `Sources/studio-cli/` | Standalone terminal bake-off | +| `Sources/SwiftPythonKit/` | Vendored V2 pool bindings (generated by `swift-python-gen --surface pool`) | +| `scripts/build_app.sh` | Assemble `.app` (`--bundle-python`, `--sign`, `--open`) | +| `scripts/bundle_python.sh` | Bundle a self-contained Python + scientific stack | +| `SHIPPING.md` | Notarized DMG runbook (download-and-run for anyone) | + +## Distribution + +Studio is meant to ship as a notarized, stapled DMG with a **self-contained Python** +(no system install needed) — see [`SHIPPING.md`](SHIPPING.md). The `IrisDemo` +sample remains the minimal "hello world"; Studio is the flagship on the V2 surface. diff --git a/Examples/Studio/Resources/AppIcon-1024.png b/Examples/Studio/Resources/AppIcon-1024.png new file mode 100644 index 0000000..bbba6e8 Binary files /dev/null and b/Examples/Studio/Resources/AppIcon-1024.png differ diff --git a/Examples/Studio/Resources/AppIcon-source.png b/Examples/Studio/Resources/AppIcon-source.png new file mode 100644 index 0000000..74e0ca6 Binary files /dev/null and b/Examples/Studio/Resources/AppIcon-source.png differ diff --git a/Examples/Studio/Resources/AppIcon.icns b/Examples/Studio/Resources/AppIcon.icns new file mode 100644 index 0000000..10e84c9 Binary files /dev/null and b/Examples/Studio/Resources/AppIcon.icns differ diff --git a/Examples/Studio/SHIPPING.md b/Examples/Studio/SHIPPING.md new file mode 100644 index 0000000..23cdc68 --- /dev/null +++ b/Examples/Studio/SHIPPING.md @@ -0,0 +1,134 @@ +# Shipping SwiftPython Studio + +How to turn `SwiftPython Studio.app` into a notarized, stapled DMG that anyone can +download and run with **no Python installed**. This is a standard macOS Developer ID +notarization flow with the extra steps a bundled-Python app needs. + +## What ships + +- A native macOS app built on the **commercial `SwiftPythonRuntime.xcframework`** + (v0.5.14) — the exact artifact a customer consumes. +- The bundled **`SwiftPythonWorker`** sidecar (auto-discovered by `PythonProcessPool`). +- A **bundled, relocatable Python 3.13 + numpy/pandas/scikit-learn/scipy** so the + app is self-contained (the IRIS sample, by contrast, requires Homebrew Python). +- The canonical V2 `pool.` bindings, vendored as `SwiftPythonKit`. +- The `studio-cli` terminal demo (in `Contents/MacOS/`, also droppable in the DMG). + +## Prerequisites (release machine) + +- A `Developer ID Application` identity in the login Keychain + (`security find-identity -v -p codesigning`). +- A notary keychain profile created once with: + ```bash + xcrun notarytool store-credentials "" \ + --apple-id "" --team-id "" --password "" + ``` + The commands below use `"$NOTARY_PROFILE"`; export it to your profile name. +- Homebrew `python@3.13` (source for the relocatable bundle) with the four wheels. + +## Build → sign → notarize → DMG + +```bash +cd Examples/Studio +export NOTARY_PROFILE="" + +# 1. Assemble, bundle Python, and sign (Developer ID, hardened runtime, inside-out). +STUDIO_APP_SHORT_VERSION=0.1.0 STUDIO_APP_BUILD_VERSION=1 \ + ./scripts/build_app.sh --bundle-python --sign + +# 2. Notarize the app, then staple. +APP="build/SwiftPython Studio.app" +ditto -c -k --keepParent "$APP" build/Studio.zip +xcrun notarytool submit build/Studio.zip --keychain-profile "$NOTARY_PROFILE" --wait +xcrun stapler staple "$APP" +spctl -a -vv --type execute "$APP" # → "Notarized Developer ID" + +# 3. Build a DMG (with an Applications drop-target), sign + notarize + staple it. +rm -rf build/dmg && mkdir build/dmg +cp -R "$APP" build/dmg/ && ln -s /Applications build/dmg/Applications +hdiutil create -volname "SwiftPython Studio" -srcfolder build/dmg -ov -format UDZO build/SwiftPythonStudio.dmg +codesign --force --sign "Developer ID Application" build/SwiftPythonStudio.dmg +xcrun notarytool submit build/SwiftPythonStudio.dmg --keychain-profile "$NOTARY_PROFILE" --wait +xcrun stapler staple build/SwiftPythonStudio.dmg +``` + +## Signing order (load-bearing) + +`build_app.sh --sign` signs **inside-out**, which is mandatory or notarization +returns `Invalid`: + +1. Every nested Mach-O (wheel `.so`/`.dylib`, framework dylibs, the Python binary) + under `Contents/Frameworks` + `Contents/Resources/Python` + (Developer ID + `--options runtime` + `--timestamp`). +2. Non-Mach-O Python payloads with executable bits are stripped (`chmod -x`) or a + quarantined launch shows **"App is damaged"** even when `spctl` passes. +3. The `Python.framework` bundle (re-sealed). +4. The `SwiftPythonWorker` sidecar + `studio-cli`. +5. The outer `.app` last (no `--deep`), with `scripts/studio.entitlements` + (`disable-library-validation`, `allow-dyld-environment-variables`). + +## Wiring the worker to the bundled interpreter + +The shipped `SwiftPythonWorker` must use the **bundled** Python, not a system one. +Two pieces, both implemented: + +- **Relink** the three binaries (`Studio`, `studio-cli`, `SwiftPythonWorker`) from + the absolute Homebrew Python load path to `@rpath/Python.framework/...` with an + added rpath of `@executable_path/../Frameworks` (done in `build_app.sh --bundle-python`). + Re-sign after this (`install_name_tool` invalidates the signature). +- **Env**: `StudioRuntime.configureBundledPythonIfPresent()` sets, relative to the + running executable, `PYTHONHOME = …/Contents/Frameworks/Python.framework/Versions/3.13` + and `PYTHONPATH = …/Contents/Resources/Python/site-packages` before the pool spawns + the worker (which inherits them). `studio.entitlements` enables + `allow-dyld-environment-variables` so this is honored under the hardened runtime. + +Verify on a Mac with **no Homebrew / no Python** (or simulate locally with +`env -i HOME="$HOME" "$APP/Contents/MacOS/studio-cli"`): + +```bash +xcrun stapler validate "build/SwiftPython Studio.app" +# also launch from Finder (not Terminal) — the GUI launchd env has no Homebrew PATH, +# so this proves the bundled interpreter + env wiring are what make it work for strangers. +``` + +## Hosting + +- Upload `SwiftPythonStudio.dmg` to a **public GitHub release** (a dedicated + releases repo, or this repo's Releases). +- Link to it from your download/marketing page. +- Sparkle auto-update is **deferred** for v1 (static DMG). To add it later, wire a + standard Sparkle appcast under a `studio/` feed path; nothing in the build here + changes. + +## Gotchas hit while producing the first notarized DMG (do not repeat) + +1. **Relinked binaries must be re-signed.** `install_name_tool` (used to point the + 3 binaries at `@rpath/Python.framework`) invalidates the signature; on Apple + Silicon the binary then fails to launch with `Bad executable`. The `--sign` + step runs *after* relink for this reason. +2. **Do NOT sign the app's main executable separately.** Signing + `Contents/MacOS/Studio` directly *and* sealing the `.app` yields "The signature + of the binary is invalid" at notarization. Sign only the `.app` (it signs the + main executable + applies entitlements). Helper executables (`SwiftPythonWorker`, + `studio-cli`) DO need their own `codesign`. +3. **Sign the framework as a bundle**, not its `Versions/3.13` directory: + `codesign … Contents/Frameworks/Python.framework`. +4. **Strip symlinks that escape the framework.** Homebrew's framework links + `lib/python3.13/site-packages` up into the Cellar; Gatekeeper rejects it with + "invalid destination for symbolic link in bundle." `bundle_python.sh` removes + escaping/absolute symlinks (we use `Contents/Resources/Python/site-packages` + via `PYTHONPATH` anyway). + +## Status — DONE (first release built) + +The full path is implemented, run, and verified on the build machine: + +- `numpy 2.5 / pandas 3.0 / scikit-learn 1.9 / scipy 1.18` bundled (lean tier). +- The bake-off runs in a **fully sanitized environment** (`env -i`, no Homebrew, + no inherited `PYTHONHOME`) — proof it works on a Python-less Mac. +- App: notarized (Accepted) + stapled + `spctl … → "Notarized Developer ID"`. +- **`build/SwiftPythonStudio.dmg`** (~182 MB): notarized + stapled + Gatekeeper + accepted (`source=Notarized Developer ID`). + +Remaining before public download: host the DMG on a public GitHub release + link it +from the download page, and one real download-and-run smoke on a *different* Mac. diff --git a/Examples/Studio/Sources/Studio/BakeOffViewModel.swift b/Examples/Studio/Sources/Studio/BakeOffViewModel.swift new file mode 100644 index 0000000..bd2472b --- /dev/null +++ b/Examples/Studio/Sources/Studio/BakeOffViewModel.swift @@ -0,0 +1,217 @@ +import Foundation +import SwiftUI +import SwiftPythonRuntime +import SwiftPythonKit + +@MainActor +final class BakeOffViewModel: ObservableObject { + enum Phase: Equatable { case idle, running, done } + + @Published var phase: Phase = .idle + @Published var candidates: [Candidate] + @Published var workers: [WorkerLane] + @Published var log: [LogLine] = [] + @Published var confusion: ConfusionMatrix? + @Published var winnerName: String? + @Published var elapsedMs: Int = 0 + + let dataset: DatasetInfo + private let classNames = ["Setosa", "Vinifera", "Acuminata"] + private let workerCount = 4 + private var pool: PythonProcessPool? + + // Cached dataset (built once, deterministic). + private let xTrain: [[Double]] + private let yTrain: [Int] + private let xTest: [[Double]] + private let yTest: [Int] + + init() { + let (xtr, ytr, xte, yte) = Self.makeDataset(samples: 1500, features: 12, classes: 3, seed: 0xC0FFEE) + self.xTrain = xtr; self.yTrain = ytr; self.xTest = xte; self.yTest = yte + self.dataset = DatasetInfo(samples: 1500, features: 12, classes: 3, trainCount: ytr.count, testCount: yte.count) + + self.candidates = [ + Candidate(name: "Random Forest", estimator: .randomForest, workerIndex: 0, colorIndex: 0, + code: "pool.sklearn.ensemble\n .RandomForestClassifier(n_estimators: 300)"), + Candidate(name: "Gradient Boosting", estimator: .gradientBoosting, workerIndex: 1, colorIndex: 1, + code: "pool.sklearn.ensemble\n .GradientBoostingClassifier(n_estimators: 200)"), + Candidate(name: "Logistic Regression", estimator: .logisticRegression, workerIndex: 2, colorIndex: 2, + code: "pool.sklearn.linearmodel\n .LogisticRegression(max_iter: 2000)"), + Candidate(name: "K-Neighbors", estimator: .kNeighbors, workerIndex: 3, colorIndex: 3, + code: "pool.sklearn.neighbors\n .KNeighborsClassifier(n_neighbors: 7)"), + ] + self.workers = (0.. String? { + let fm = FileManager.default + if let env = ProcessInfo.processInfo.environment["SWIFTPYTHON_WORKER_PATH"], !env.isEmpty, + fm.fileExists(atPath: env) { return env } + if let dir = Bundle.main.executableURL?.deletingLastPathComponent() { + let bundled = dir.appendingPathComponent("SwiftPythonWorker").path + if fm.fileExists(atPath: bundled) { return bundled } + } + return nil // fall back to runtime auto-discovery + } + + // MARK: - Deterministic dataset + + /// Gaussian clusters per class → a real multi-class classification problem with + /// enough overlap to spread model accuracy. Fully deterministic for reproducible demos. + private static func makeDataset(samples: Int, features: Int, classes: Int, seed: UInt64) + -> ([[Double]], [Int], [[Double]], [Int]) { + var rng = SplitMix64(seed: seed) + // Class centroids spaced across feature space. + var centroids: [[Double]] = [] + for c in 0.. ConfusionMatrix { + let n = labels.count + var counts = Array(repeating: Array(repeating: 0, count: n), count: n) + for (t, p) in zip(truth, pred) where t >= 0 && t < n && p >= 0 && p < n { + counts[t][p] += 1 + } + return ConfusionMatrix(labels: labels, counts: counts) + } +} + +/// Tiny deterministic PRNG (SplitMix64) — no Foundation randomness, fully reproducible. +private struct SplitMix64 { + private var state: UInt64 + init(seed: UInt64) { state = seed } + mutating func next() -> UInt64 { + state &+= 0x9E3779B97F4A7C15 + var z = state + z = (z ^ (z >> 30)) &* 0xBF58476D1CE4E5B9 + z = (z ^ (z >> 27)) &* 0x94D049BB133111EB + return z ^ (z >> 31) + } + mutating func nextUnit() -> Double { Double(next() >> 11) * (1.0 / 9007199254740992.0) } + mutating func nextGaussian() -> Double { + let u1 = max(nextUnit(), 1e-12), u2 = nextUnit() + return (-2.0 * log(u1)).squareRoot() * cos(2.0 * .pi * u2) + } +} diff --git a/Examples/Studio/Sources/Studio/Models.swift b/Examples/Studio/Sources/Studio/Models.swift new file mode 100644 index 0000000..47120f7 --- /dev/null +++ b/Examples/Studio/Sources/Studio/Models.swift @@ -0,0 +1,115 @@ +import Foundation +import SwiftPythonRuntime +import SwiftPythonKit + +// MARK: - Candidate + +enum RunStatus: Equatable { + case idle + case training + case done +} + +/// One scikit-learn classifier raced in the bake-off, pinned to one worker. +struct Candidate: Identifiable { + let id = UUID() + let name: String + let estimator: EstimatorKind + let workerIndex: Int + let colorIndex: Int + /// The literal Swift the app runs for this candidate — shown on screen. + let code: String + + var status: RunStatus = .idle + var accuracy: Double? = nil + var elapsedMs: Int? = nil +} + +/// Which generated `pool.sklearn` estimator a candidate drives. Each case knows how +/// to train + score + predict through the canonical pool surface. +enum EstimatorKind { + case randomForest + case gradientBoosting + case logisticRegression + case kNeighbors + + /// Train + score on the pinned worker, returning (testAccuracy, testPredictions). + func trainScorePredict( + pool: PythonProcessPool, + worker: Int, + xTrain: [[Double]], yTrain: [Int], + xTest: [[Double]], yTest: [Int] + ) async throws -> (accuracy: Double, predictions: [Int]) { + switch self { + case .randomForest: + let m = try await pool.sklearn.ensemble.RandomForestClassifier(n_estimators: 300, random_state: 0, worker: worker) + let fitted = try await m.fit(X: xTrain, y: yTrain, worker: worker) + let acc = try await fitted.score(X: xTest, y: yTest, worker: worker) + let preds = try await Self.predictList(pool: pool, handle: fitted, x: xTest, worker: worker) + return (acc, preds) + case .gradientBoosting: + let m = try await pool.sklearn.ensemble.GradientBoostingClassifier(n_estimators: 200, random_state: 0, worker: worker) + let fitted = try await m.fit(X: xTrain, y: yTrain, worker: worker) + let acc = try await fitted.score(X: xTest, y: yTest, worker: worker) + let preds = try await Self.predictList(pool: pool, handle: fitted, x: xTest, worker: worker) + return (acc, preds) + case .logisticRegression: + let m = try await pool.sklearn.linearmodel.LogisticRegression(random_state: 0, max_iter: 2000, worker: worker) + let fitted = try await m.fit(X: xTrain, y: yTrain, worker: worker) + let acc = try await fitted.score(X: xTest, y: yTest, worker: worker) + let preds = try await Self.predictList(pool: pool, handle: fitted, x: xTest, worker: worker) + return (acc, preds) + case .kNeighbors: + let m = try await pool.sklearn.neighbors.KNeighborsClassifier(n_neighbors: 7, worker: worker) + let fitted = try await m.fit(X: xTrain, y: yTrain, worker: worker) + let acc = try await fitted.score(X: xTest, y: yTest, worker: worker) + let preds = try await Self.predictList(pool: pool, handle: fitted, x: xTest, worker: worker) + return (acc, preds) + } + } + + private static func predictList( + pool: PythonProcessPool, + handle: some HandleConvertible, + x: [[Double]], + worker: Int + ) async throws -> [Int] { + let predsHandle = try await pool.methodOwned(handle: handle, name: "predict", args: [.python(x)], worker: worker) + return try await pool.methodResult(handle: predsHandle, name: "tolist", worker: worker) + } +} + +// MARK: - Worker lane + +struct WorkerLane: Identifiable { + let id: Int // worker index + var label: String = "idle" + var status: RunStatus = .idle + var colorIndex: Int? = nil +} + +// MARK: - Console + +struct LogLine: Identifiable { + let id = UUID() + let text: String + let kind: Kind + enum Kind { case prompt, info, result, dim } +} + +// MARK: - Confusion matrix + +struct ConfusionMatrix { + let labels: [String] + let counts: [[Int]] // counts[true][pred] +} + +// MARK: - Dataset + +struct DatasetInfo { + let samples: Int + let features: Int + let classes: Int + let trainCount: Int + let testCount: Int +} diff --git a/Examples/Studio/Sources/Studio/StudioApp.swift b/Examples/Studio/Sources/Studio/StudioApp.swift new file mode 100644 index 0000000..9d16196 --- /dev/null +++ b/Examples/Studio/Sources/Studio/StudioApp.swift @@ -0,0 +1,12 @@ +import SwiftUI + +@main +struct StudioApp: App { + var body: some Scene { + WindowGroup { + ContentView() + } + .windowResizability(.contentSize) + .windowStyle(.hiddenTitleBar) + } +} diff --git a/Examples/Studio/Sources/Studio/Theme.swift b/Examples/Studio/Sources/Studio/Theme.swift new file mode 100644 index 0000000..db63d5b --- /dev/null +++ b/Examples/Studio/Sources/Studio/Theme.swift @@ -0,0 +1,27 @@ +import SwiftUI + +/// Dark, high-contrast palette tuned for screen capture (square 1:1 in-feed video). +enum Theme { + static let bg = Color(red: 0.05, green: 0.06, blue: 0.09) + static let panel = Color(red: 0.09, green: 0.10, blue: 0.14) + static let panelStroke = Color.white.opacity(0.07) + static let text = Color(red: 0.92, green: 0.94, blue: 0.98) + static let dim = Color(red: 0.55, green: 0.58, blue: 0.66) + static let accent = Color(red: 1.0, green: 0.58, blue: 0.0) // Swift orange + static let python = Color(red: 0.21, green: 0.49, blue: 0.74) // Python blue + static let good = Color(red: 0.30, green: 0.85, blue: 0.55) + static let busy = Color(red: 1.0, green: 0.78, blue: 0.25) + static let idle = Color.white.opacity(0.12) + static let term = Color(red: 0.02, green: 0.03, blue: 0.05) + + static let mono = Font.system(.body, design: .monospaced) + static let monoSmall = Font.system(size: 12, design: .monospaced) + static let title = Font.system(size: 30, weight: .heavy, design: .rounded) + + static let modelColors: [Color] = [ + Color(red: 1.0, green: 0.58, blue: 0.0), + Color(red: 0.36, green: 0.74, blue: 1.0), + Color(red: 0.62, green: 0.45, blue: 1.0), + Color(red: 0.30, green: 0.85, blue: 0.55), + ] +} diff --git a/Examples/Studio/Sources/Studio/Views.swift b/Examples/Studio/Sources/Studio/Views.swift new file mode 100644 index 0000000..874ec75 --- /dev/null +++ b/Examples/Studio/Sources/Studio/Views.swift @@ -0,0 +1,271 @@ +import SwiftUI +import Charts + +struct ContentView: View { + @StateObject private var vm = BakeOffViewModel() + + var body: some View { + VStack(spacing: 14) { + header + HStack(alignment: .top, spacing: 14) { + codePanel.frame(width: 420) + resultsPanel + } + workerStrip + ConsoleView(lines: vm.log) + .frame(maxWidth: .infinity, minHeight: 190, maxHeight: 210) + footer + } + .padding(22) + .frame(width: 1000, height: 1000) + .background(Theme.bg) + .preferredColorScheme(.dark) + .task { + // Optional auto-run for screen capture / demos. + if ProcessInfo.processInfo.environment["STUDIO_AUTORUN"] == "1" { + try? await Task.sleep(nanoseconds: 1_200_000_000) + vm.run() + } + } + } + + // MARK: Header + + private var header: some View { + HStack(alignment: .center) { + VStack(alignment: .leading, spacing: 2) { + Text("SwiftPython Studio").font(Theme.title).foregroundStyle(Theme.text) + Text("Swift driving scikit-learn — across real Python processes, live.") + .font(.system(size: 14)).foregroundStyle(Theme.dim) + } + Spacer() + if vm.elapsedMs > 0 { + badge("\(vm.elapsedMs) ms", color: Theme.python) + } + Button(action: { vm.run() }) { + HStack(spacing: 8) { + Image(systemName: vm.phase == .running ? "hourglass" : "play.fill") + Text(vm.phase == .running ? "Racing…" : "Run bake-off") + } + .font(.system(size: 16, weight: .bold, design: .rounded)) + .padding(.horizontal, 18).padding(.vertical, 11) + .background(Theme.accent).foregroundStyle(.black) + .clipShape(Capsule()) + } + .buttonStyle(.plain) + .disabled(vm.phase == .running) + .opacity(vm.phase == .running ? 0.6 : 1) + } + } + + // MARK: Live code + + private var codePanel: some View { + panel("Live Swift") { + VStack(alignment: .leading, spacing: 10) { + ForEach(vm.candidates) { c in + VStack(alignment: .leading, spacing: 6) { + HStack(spacing: 8) { + statusDot(c.status, color: Theme.modelColors[c.colorIndex]) + Text(c.name).font(.system(size: 14, weight: .semibold)).foregroundStyle(Theme.text) + Spacer() + Text("worker \(c.workerIndex)").font(Theme.monoSmall).foregroundStyle(Theme.dim) + } + Text(c.code) + .font(Theme.monoSmall) + .foregroundStyle(Theme.text.opacity(0.92)) + .padding(10) + .frame(maxWidth: .infinity, alignment: .leading) + .background(Theme.term) + .clipShape(RoundedRectangle(cornerRadius: 8)) + } + } + } + } + } + + // MARK: Results + + private var resultsPanel: some View { + panel("Test accuracy") { + VStack(alignment: .leading, spacing: 14) { + Chart(vm.candidates) { c in + BarMark( + x: .value("Accuracy", c.accuracy ?? 0), + y: .value("Model", c.name) + ) + .foregroundStyle(Theme.modelColors[c.colorIndex]) + .annotation(position: .trailing) { + if let a = c.accuracy { + Text(String(format: "%.3f", a)).font(Theme.monoSmall).foregroundStyle(Theme.text) + } + } + .cornerRadius(4) + } + .chartXScale(domain: 0...1) + .chartXAxis { AxisMarks(values: [0, 0.25, 0.5, 0.75, 1.0]) } + .frame(height: 180) + .animation(.easeOut(duration: 0.5), value: vm.candidates.map(\.accuracy)) + + if let cm = vm.confusion { + ConfusionView(matrix: cm, winner: vm.winnerName) + } else { + Text("Run the bake-off to race four models in parallel.") + .font(.system(size: 13)).foregroundStyle(Theme.dim) + .frame(maxWidth: .infinity, minHeight: 150, alignment: .center) + } + } + } + } + + // MARK: Worker strip + + private var workerStrip: some View { + panel("Python workers") { + HStack(spacing: 10) { + ForEach(vm.workers) { w in + HStack(spacing: 8) { + statusDot(w.status, color: w.colorIndex.map { Theme.modelColors[$0] } ?? Theme.idle) + VStack(alignment: .leading, spacing: 1) { + Text("worker \(w.id)").font(.system(size: 12, weight: .bold)).foregroundStyle(Theme.text) + Text(w.status == .idle ? "idle" : w.label) + .font(Theme.monoSmall).foregroundStyle(Theme.dim).lineLimit(1) + } + Spacer() + } + .padding(10) + .frame(maxWidth: .infinity) + .background(Theme.term) + .clipShape(RoundedRectangle(cornerRadius: 8)) + .overlay(RoundedRectangle(cornerRadius: 8).stroke( + w.status == .training ? Theme.busy.opacity(0.7) : Theme.panelStroke, lineWidth: 1)) + } + } + } + } + + private var footer: some View { + HStack { + Text("SwiftPythonRuntime 0.5.14").font(Theme.monoSmall).foregroundStyle(Theme.dim) + Spacer() + Text("pool. · zero-copy handles · multi-process") + .font(Theme.monoSmall).foregroundStyle(Theme.dim) + } + } + + // MARK: Building blocks + + private func panel(_ title: String, @ViewBuilder _ content: () -> C) -> some View { + VStack(alignment: .leading, spacing: 10) { + Text(title.uppercased()) + .font(.system(size: 11, weight: .bold)).tracking(1.2).foregroundStyle(Theme.dim) + content() + } + .padding(16) + .frame(maxWidth: .infinity, alignment: .leading) + .background(Theme.panel) + .clipShape(RoundedRectangle(cornerRadius: 14)) + .overlay(RoundedRectangle(cornerRadius: 14).stroke(Theme.panelStroke, lineWidth: 1)) + } + + private func badge(_ text: String, color: Color) -> some View { + Text(text).font(.system(size: 13, weight: .bold, design: .monospaced)) + .padding(.horizontal, 12).padding(.vertical, 7) + .background(color.opacity(0.18)).foregroundStyle(color) + .clipShape(Capsule()) + } + + private func statusDot(_ status: RunStatus, color: Color) -> some View { + Circle() + .fill(status == .idle ? Theme.idle : (status == .done ? Theme.good : color)) + .frame(width: 10, height: 10) + .overlay(Circle().stroke(.white.opacity(0.15), lineWidth: 1)) + .opacity(status == .training ? 0.5 : 1) + .animation(status == .training ? .easeInOut(duration: 0.7).repeatForever(autoreverses: true) : .default, + value: status) + } +} + +// MARK: - Confusion matrix + +struct ConfusionView: View { + let matrix: ConfusionMatrix + let winner: String? + + private var maxCount: Int { matrix.counts.flatMap { $0 }.max() ?? 1 } + + var body: some View { + VStack(alignment: .leading, spacing: 8) { + HStack { + Text("Confusion matrix").font(.system(size: 12, weight: .semibold)).foregroundStyle(Theme.text) + if let winner { Text("· \(winner)").font(Theme.monoSmall).foregroundStyle(Theme.good) } + } + Grid(horizontalSpacing: 4, verticalSpacing: 4) { + ForEach(matrix.counts.indices, id: \.self) { r in + GridRow { + ForEach(matrix.counts[r].indices, id: \.self) { c in + let v = matrix.counts[r][c] + Text("\(v)") + .font(Theme.monoSmall) + .frame(width: 54, height: 38) + .background(Theme.good.opacity(Double(v) / Double(maxCount) * 0.75 + (r == c ? 0.12 : 0))) + .foregroundStyle(Theme.text) + .clipShape(RoundedRectangle(cornerRadius: 6)) + } + } + } + } + Text("rows = true class · columns = predicted") + .font(.system(size: 11)).foregroundStyle(Theme.dim) + } + } +} + +// MARK: - Terminal-style console ("behind the curtain") + +struct ConsoleView: View { + let lines: [LogLine] + + var body: some View { + ScrollViewReader { proxy in + ScrollView { + VStack(alignment: .leading, spacing: 3) { + ForEach(lines) { line in + HStack(alignment: .top, spacing: 6) { + Text(prefix(line.kind)).foregroundStyle(color(line.kind)).font(Theme.monoSmall) + Text(line.text).foregroundStyle(color(line.kind)).font(Theme.monoSmall) + .frame(maxWidth: .infinity, alignment: .leading) + } + .id(line.id) + } + Color.clear.frame(height: 1).id("bottom") + } + .padding(12) + .frame(maxWidth: .infinity, alignment: .leading) + } + .background(Theme.term) + .clipShape(RoundedRectangle(cornerRadius: 12)) + .overlay(RoundedRectangle(cornerRadius: 12).stroke(Theme.panelStroke, lineWidth: 1)) + .onChange(of: lines.count) { _, _ in + withAnimation { proxy.scrollTo("bottom", anchor: .bottom) } + } + } + } + + private func prefix(_ k: LogLine.Kind) -> String { + switch k { + case .prompt: return "›" + case .result: return "✓" + case .info: return "•" + case .dim: return " " + } + } + private func color(_ k: LogLine.Kind) -> Color { + switch k { + case .prompt: return Theme.accent + case .result: return Theme.good + case .info: return Theme.text.opacity(0.85) + case .dim: return Theme.dim + } + } +} diff --git a/Examples/Studio/Sources/SwiftPythonKit/Bootstrap.swift b/Examples/Studio/Sources/SwiftPythonKit/Bootstrap.swift new file mode 100644 index 0000000..3f514eb --- /dev/null +++ b/Examples/Studio/Sources/SwiftPythonKit/Bootstrap.swift @@ -0,0 +1,30 @@ +import Foundation + +/// Runtime helpers for the Studio app/CLI (hand-written; not generated). +public enum StudioRuntime { + /// When a bundled Python framework is present (i.e. running from a packaged + /// `.app` produced with `build_app.sh --bundle-python`), point the interpreter + /// at the bundled stdlib + site-packages so the worker uses the bundled Python + /// instead of a system/Homebrew one. The worker inherits these env vars as a + /// child process. No-op in dev (`swift run`), where the system Python is used. + public static func configureBundledPythonIfPresent() { + // Resolve relative to the running executable (works for both the app's main + // binary and a helper CLI inside Contents/MacOS), not Bundle.main.bundleURL + // which differs between the two. + guard let exeDir = Bundle.main.executableURL?.resolvingSymlinksInPath().deletingLastPathComponent() else { + return + } + let contents = exeDir.deletingLastPathComponent() // .../Contents + let versionDir = contents.appendingPathComponent("Frameworks/Python.framework/Versions/3.13") + guard FileManager.default.fileExists(atPath: versionDir.path) else { return } + + setenv("PYTHONHOME", versionDir.path, 1) + + let site = contents.appendingPathComponent("Resources/Python/site-packages").path + if let existing = ProcessInfo.processInfo.environment["PYTHONPATH"], !existing.isEmpty { + setenv("PYTHONPATH", site + ":" + existing, 1) + } else { + setenv("PYTHONPATH", site, 1) + } + } +} diff --git a/Examples/Studio/Sources/SwiftPythonKit/PoolNumPy.swift b/Examples/Studio/Sources/SwiftPythonKit/PoolNumPy.swift new file mode 100644 index 0000000..95d95f0 --- /dev/null +++ b/Examples/Studio/Sources/SwiftPythonKit/PoolNumPy.swift @@ -0,0 +1,7921 @@ +// Generated by swift-python-gen +// Do not edit manually + +import SwiftPythonRuntime +import Foundation + +// MARK: - NumPy + +extension PythonProcessPool { + /// Pool-bound `numpy` namespace. Calls run on this pool. + public var numpy: PoolNumPy { PoolNumPy(pool: self) } +} + +public struct PoolNumPy: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func array(object: any PythonConvertible, dtype: String? = nil, copy: Bool? = nil, order: String? = nil, ndmin: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(object)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let ndmin = ndmin { + __kwargs["ndmin"] = RemotePythonValue(ndmin) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "array", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "array", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func zeros(shape: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "zeros", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "zeros", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ones(shape: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ones", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ones", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func empty(shape: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "empty", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "empty", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func full(shape: any PythonConvertible, fill_value: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + __args.append(RemotePythonValue(fill_value)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "full", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "full", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func zeros_like(a: any PythonConvertible, dtype: String? = nil, order: String? = nil, shape: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let shape = shape { + __kwargs["shape"] = RemotePythonValue(shape) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "zeros_like", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "zeros_like", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ones_like(a: any PythonConvertible, dtype: String? = nil, order: String? = nil, shape: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let shape = shape { + __kwargs["shape"] = RemotePythonValue(shape) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ones_like", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ones_like", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func empty_like(a: any PythonConvertible, dtype: String? = nil, order: String? = nil, shape: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let shape = shape { + __kwargs["shape"] = RemotePythonValue(shape) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "empty_like", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "empty_like", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func full_like(a: any PythonConvertible, fill_value: any PythonConvertible, dtype: String? = nil, order: String? = nil, shape: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(fill_value)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let shape = shape { + __kwargs["shape"] = RemotePythonValue(shape) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "full_like", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "full_like", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arange(start: Int, stop: Int? = nil, step: Int? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(start)) + var __kwargs: [String: RemotePythonValue] = [:] + if let stop = stop { + __kwargs["stop"] = RemotePythonValue(stop) + } + if let step = step { + __kwargs["step"] = RemotePythonValue(step) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arange", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arange", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func linspace(start: Double, stop: Double, num: Int? = nil, endpoint: Bool? = nil, retstep: Bool? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(start)) + __args.append(RemotePythonValue(stop)) + var __kwargs: [String: RemotePythonValue] = [:] + if let num = num { + __kwargs["num"] = RemotePythonValue(num) + } + if let endpoint = endpoint { + __kwargs["endpoint"] = RemotePythonValue(endpoint) + } + if let retstep = retstep { + __kwargs["retstep"] = RemotePythonValue(retstep) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "linspace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "linspace", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logspace(start: Double, stop: Double, num: Int? = nil, endpoint: Bool? = nil, base: Double? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(start)) + __args.append(RemotePythonValue(stop)) + var __kwargs: [String: RemotePythonValue] = [:] + if let num = num { + __kwargs["num"] = RemotePythonValue(num) + } + if let endpoint = endpoint { + __kwargs["endpoint"] = RemotePythonValue(endpoint) + } + if let base = base { + __kwargs["base"] = RemotePythonValue(base) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logspace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logspace", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func geomspace(start: Double, stop: Double, num: Int? = nil, endpoint: Bool? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(start)) + __args.append(RemotePythonValue(stop)) + var __kwargs: [String: RemotePythonValue] = [:] + if let num = num { + __kwargs["num"] = RemotePythonValue(num) + } + if let endpoint = endpoint { + __kwargs["endpoint"] = RemotePythonValue(endpoint) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "geomspace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "geomspace", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func eye(N: Int, M: Int? = nil, k: Int? = nil, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(N)) + var __kwargs: [String: RemotePythonValue] = [:] + if let M = M { + __kwargs["M"] = RemotePythonValue(M) + } + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "eye", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "eye", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func identity(n: Int, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "identity", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "identity", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func diag(v: any PythonConvertible, k: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "diag", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "diag", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func diagflat(v: any PythonConvertible, k: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "diagflat", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "diagflat", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tri(N: Int, M: Int? = nil, k: Int? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(N)) + var __kwargs: [String: RemotePythonValue] = [:] + if let M = M { + __kwargs["M"] = RemotePythonValue(M) + } + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "tri", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "tri", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tril(m: any PythonConvertible, k: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "tril", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "tril", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func triu(m: any PythonConvertible, k: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "triu", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "triu", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func vander(x: any PythonConvertible, N: Int? = nil, increasing: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let N = N { + __kwargs["N"] = RemotePythonValue(N) + } + if let increasing = increasing { + __kwargs["increasing"] = RemotePythonValue(increasing) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "vander", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "vander", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func meshgrid(_ args: [RemotePythonValue] = [], copy: Bool? = nil, sparse: Bool? = nil, indexing: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let sparse = sparse { + __kwargs["sparse"] = RemotePythonValue(sparse) + } + if let indexing = indexing { + __kwargs["indexing"] = RemotePythonValue(indexing) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "meshgrid", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "meshgrid", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fromfunction(function: any PythonConvertible, shape: any PythonConvertible, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(function)) + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fromfunction", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fromfunction", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fromiter(iter: any PythonConvertible, dtype: String, count: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(iter)) + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let count = count { + __kwargs["count"] = RemotePythonValue(count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fromiter", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fromiter", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func frombuffer(buffer: any PythonConvertible, dtype: String? = nil, count: Int? = nil, offset: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(buffer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let count = count { + __kwargs["count"] = RemotePythonValue(count) + } + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "frombuffer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "frombuffer", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func reshape(a: any PythonConvertible, newshape: any PythonConvertible, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(newshape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "reshape", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "reshape", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ravel(a: any PythonConvertible, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ravel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ravel", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func flatten(a: any PythonConvertible, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "flatten", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "flatten", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func transpose(a: any PythonConvertible, axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "transpose", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "transpose", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func swapaxes(a: any PythonConvertible, axis1: Int, axis2: Int, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(axis1)) + __args.append(RemotePythonValue(axis2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "swapaxes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "swapaxes", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func moveaxis(a: any PythonConvertible, source: any PythonConvertible, destination: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(source)) + __args.append(RemotePythonValue(destination)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "moveaxis", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "moveaxis", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rollaxis(a: any PythonConvertible, axis: Int, start: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(axis)) + var __kwargs: [String: RemotePythonValue] = [:] + if let start = start { + __kwargs["start"] = RemotePythonValue(start) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "rollaxis", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "rollaxis", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func atleast_1d(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "atleast_1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "atleast_1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func atleast_2d(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "atleast_2d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "atleast_2d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func atleast_3d(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "atleast_3d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "atleast_3d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func broadcast_to(array: any PythonConvertible, shape: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(array)) + __args.append(RemotePythonValue(shape)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "broadcast_to", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "broadcast_to", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func broadcast_arrays(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "broadcast_arrays", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "broadcast_arrays", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func expand_dims(a: any PythonConvertible, axis: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(axis)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "expand_dims", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "expand_dims", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func squeeze(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "squeeze", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "squeeze", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func concatenate(arrays: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arrays)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "concatenate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "concatenate", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func stack(arrays: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arrays)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "stack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "stack", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func vstack(tup: any PythonConvertible, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(tup)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "vstack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "vstack", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func hstack(tup: any PythonConvertible, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(tup)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "hstack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "hstack", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func dstack(tup: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(tup)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "dstack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "dstack", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func column_stack(tup: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(tup)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "column_stack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "column_stack", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func row_stack(tup: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(tup)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "row_stack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "row_stack", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func split(ary: any PythonConvertible, indices_or_sections: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ary)) + __args.append(RemotePythonValue(indices_or_sections)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func array_split(ary: any PythonConvertible, indices_or_sections: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ary)) + __args.append(RemotePythonValue(indices_or_sections)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "array_split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "array_split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func hsplit(ary: any PythonConvertible, indices_or_sections: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ary)) + __args.append(RemotePythonValue(indices_or_sections)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "hsplit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "hsplit", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func vsplit(ary: any PythonConvertible, indices_or_sections: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ary)) + __args.append(RemotePythonValue(indices_or_sections)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "vsplit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "vsplit", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func dsplit(ary: any PythonConvertible, indices_or_sections: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ary)) + __args.append(RemotePythonValue(indices_or_sections)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "dsplit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "dsplit", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func tile(A: any PythonConvertible, reps: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(A)) + __args.append(RemotePythonValue(reps)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "tile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "tile", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func `repeat`(a: any PythonConvertible, repeats: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(repeats)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "repeat", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "repeat", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func delete(arr: any PythonConvertible, obj: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + __args.append(RemotePythonValue(obj)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "delete", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "delete", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func insert(arr: any PythonConvertible, obj: any PythonConvertible, values: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + __args.append(RemotePythonValue(obj)) + __args.append(RemotePythonValue(values)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "insert", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "insert", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func append(arr: any PythonConvertible, values: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + __args.append(RemotePythonValue(values)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "append", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "append", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func resize(a: any PythonConvertible, new_shape: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(new_shape)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "resize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "resize", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func trim_zeros(filt: any PythonConvertible, trim: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(filt)) + var __kwargs: [String: RemotePythonValue] = [:] + if let trim = trim { + __kwargs["trim"] = RemotePythonValue(trim) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "trim_zeros", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "trim_zeros", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func unique(ar: any PythonConvertible, return_index: Bool? = nil, return_inverse: Bool? = nil, return_counts: Bool? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ar)) + var __kwargs: [String: RemotePythonValue] = [:] + if let return_index = return_index { + __kwargs["return_index"] = RemotePythonValue(return_index) + } + if let return_inverse = return_inverse { + __kwargs["return_inverse"] = RemotePythonValue(return_inverse) + } + if let return_counts = return_counts { + __kwargs["return_counts"] = RemotePythonValue(return_counts) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "unique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "unique", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func flip(m: any PythonConvertible, axis: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "flip", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "flip", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fliplr(m: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fliplr", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fliplr", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func flipud(m: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "flipud", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "flipud", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rot90(m: any PythonConvertible, k: Int? = nil, axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let axes = axes { + __kwargs["axes"] = RemotePythonValue(axes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "rot90", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "rot90", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func roll(a: any PythonConvertible, shift: any PythonConvertible, axis: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(shift)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "roll", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "roll", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func pad(array: any PythonConvertible, pad_width: any PythonConvertible, mode: String? = nil, constant_values: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(array)) + __args.append(RemotePythonValue(pad_width)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let constant_values = constant_values { + __kwargs["constant_values"] = RemotePythonValue(constant_values) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "pad", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "pad", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func take(a: any PythonConvertible, indices: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, mode: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(indices)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "take", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "take", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func take_along_axis(arr: any PythonConvertible, indices: any PythonConvertible, axis: Int, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + __args.append(RemotePythonValue(indices)) + __args.append(RemotePythonValue(axis)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "take_along_axis", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "take_along_axis", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func put(a: any PythonConvertible, ind: any PythonConvertible, v: any PythonConvertible, mode: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(ind)) + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let worker { + try await pool.invokeVoid(module: "numpy", function: "put", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "put", args: __args, kwargs: __kwargs) + } + + public func put_along_axis(arr: any PythonConvertible, indices: any PythonConvertible, values: any PythonConvertible, axis: Int, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + __args.append(RemotePythonValue(indices)) + __args.append(RemotePythonValue(values)) + __args.append(RemotePythonValue(axis)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + try await pool.invokeVoid(module: "numpy", function: "put_along_axis", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "put_along_axis", args: __args, kwargs: __kwargs) + } + + public func putmask(a: any PythonConvertible, mask: any PythonConvertible, values: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(mask)) + __args.append(RemotePythonValue(values)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + try await pool.invokeVoid(module: "numpy", function: "putmask", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "putmask", args: __args, kwargs: __kwargs) + } + + public func fill_diagonal(a: any PythonConvertible, val: any PythonConvertible, wrap: Bool? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(val)) + var __kwargs: [String: RemotePythonValue] = [:] + if let wrap = wrap { + __kwargs["wrap"] = RemotePythonValue(wrap) + } + if let worker { + try await pool.invokeVoid(module: "numpy", function: "fill_diagonal", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "fill_diagonal", args: __args, kwargs: __kwargs) + } + + public func diag_indices(n: Int, ndim: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + var __kwargs: [String: RemotePythonValue] = [:] + if let ndim = ndim { + __kwargs["ndim"] = RemotePythonValue(ndim) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "diag_indices", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "diag_indices", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func diag_indices_from(arr: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "diag_indices_from", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "diag_indices_from", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func select(condlist: any PythonConvertible, choicelist: any PythonConvertible, `default`: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(condlist)) + __args.append(RemotePythonValue(choicelist)) + if let `default` = `default` { + __args.append(RemotePythonValue(`default`)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "select", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "select", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func `where`(condition: any PythonConvertible, x: (any PythonConvertible)? = nil, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(condition)) + if let x = x { + __args.append(RemotePythonValue(x)) + } + if let y = y { + __args.append(RemotePythonValue(y)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "where", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "where", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func nonzero(a: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nonzero", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nonzero", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func flatnonzero(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "flatnonzero", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "flatnonzero", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func argwhere(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "argwhere", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "argwhere", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func searchsorted(a: any PythonConvertible, v: any PythonConvertible, side: String? = nil, sorter: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let side = side { + __kwargs["side"] = RemotePythonValue(side) + } + if let sorter = sorter { + __kwargs["sorter"] = RemotePythonValue(sorter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "searchsorted", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "searchsorted", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func extract(condition: any PythonConvertible, arr: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(condition)) + __args.append(RemotePythonValue(arr)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "extract", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "extract", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func place(arr: any PythonConvertible, mask: any PythonConvertible, vals: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arr)) + __args.append(RemotePythonValue(mask)) + __args.append(RemotePythonValue(vals)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + try await pool.invokeVoid(module: "numpy", function: "place", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "place", args: __args, kwargs: __kwargs) + } + + public func compress(condition: any PythonConvertible, a: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(condition)) + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "compress", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "compress", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func add(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "add", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "add", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func subtract(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "subtract", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "subtract", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func multiply(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "multiply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "multiply", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func divide(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "divide", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "divide", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func floor_divide(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "floor_divide", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "floor_divide", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func true_divide(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "true_divide", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "true_divide", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func negative(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "negative", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "negative", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func positive(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "positive", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "positive", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func power(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "power", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "power", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func float_power(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "float_power", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "float_power", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func remainder(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "remainder", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "remainder", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func mod(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "mod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "mod", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fmod(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fmod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fmod", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func divmod(x1: any PythonConvertible, x2: any PythonConvertible, out: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + if let out = out { + __args.append(RemotePythonValue(out)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "divmod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "divmod", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func absolute(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "absolute", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "absolute", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func abs(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "abs", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "abs", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fabs(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fabs", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fabs", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rint(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "rint", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "rint", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sign(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sign", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sign", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func conj(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "conj", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "conj", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func conjugate(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "conjugate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "conjugate", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func exp(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "exp", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "exp", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func exp2(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "exp2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "exp2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func expm1(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "expm1", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "expm1", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func log(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "log", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "log", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func log2(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "log2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "log2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func log10(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "log10", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "log10", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func log1p(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "log1p", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "log1p", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logaddexp(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logaddexp", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logaddexp", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logaddexp2(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logaddexp2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logaddexp2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sqrt(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sqrt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sqrt", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func square(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "square", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "square", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func cbrt(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cbrt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cbrt", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func reciprocal(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "reciprocal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "reciprocal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func gcd(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "gcd", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "gcd", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func lcm(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "lcm", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "lcm", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sin(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sin", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func cos(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cos", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cos", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tan(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "tan", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "tan", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arcsin(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arcsin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arcsin", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arccos(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arccos", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arccos", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arctan(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arctan", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arctan", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arctan2(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arctan2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arctan2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func hypot(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "hypot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "hypot", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sinh(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sinh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sinh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func cosh(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cosh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cosh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tanh(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "tanh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "tanh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arcsinh(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arcsinh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arcsinh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arccosh(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arccosh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arccosh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func arctanh(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "arctanh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "arctanh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func degrees(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "degrees", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "degrees", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func radians(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "radians", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "radians", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func deg2rad(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "deg2rad", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "deg2rad", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rad2deg(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "rad2deg", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "rad2deg", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func unwrap(p: any PythonConvertible, discont: Double? = nil, axis: Int? = nil, period: Double? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + var __kwargs: [String: RemotePythonValue] = [:] + if let discont = discont { + __kwargs["discont"] = RemotePythonValue(discont) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let period = period { + __kwargs["period"] = RemotePythonValue(period) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "unwrap", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "unwrap", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func around(a: any PythonConvertible, decimals: Int? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let decimals = decimals { + __kwargs["decimals"] = RemotePythonValue(decimals) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "around", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "around", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func round(a: any PythonConvertible, decimals: Int? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let decimals = decimals { + __kwargs["decimals"] = RemotePythonValue(decimals) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "round", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "round", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func floor(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "floor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "floor", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ceil(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ceil", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ceil", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func trunc(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "trunc", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "trunc", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fix(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fix", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sum(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sum", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func prod(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "prod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "prod", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nansum(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nansum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nansum", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanprod(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanprod", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cumsum(a: any PythonConvertible, axis: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cumsum", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func cumprod(a: any PythonConvertible, axis: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cumprod", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func nancumsum(a: any PythonConvertible, axis: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nancumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nancumsum", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func nancumprod(a: any PythonConvertible, axis: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nancumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nancumprod", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func diff(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, prepend: (any PythonConvertible)? = nil, append: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let prepend = prepend { + __kwargs["prepend"] = RemotePythonValue(prepend) + } + if let append = append { + __kwargs["append"] = RemotePythonValue(append) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "diff", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "diff", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ediff1d(ary: any PythonConvertible, to_end: (any PythonConvertible)? = nil, to_begin: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ary)) + if let to_end = to_end { + __args.append(RemotePythonValue(to_end)) + } + if let to_begin = to_begin { + __args.append(RemotePythonValue(to_begin)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ediff1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ediff1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func gradient(_ args: [RemotePythonValue] = [], f: any PythonConvertible, axis: (any PythonConvertible)? = nil, edge_order: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(f)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let edge_order = edge_order { + __kwargs["edge_order"] = RemotePythonValue(edge_order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "gradient", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "gradient", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cross(a: any PythonConvertible, b: any PythonConvertible, axisa: Int? = nil, axisb: Int? = nil, axisc: Int? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axisa = axisa { + __kwargs["axisa"] = RemotePythonValue(axisa) + } + if let axisb = axisb { + __kwargs["axisb"] = RemotePythonValue(axisb) + } + if let axisc = axisc { + __kwargs["axisc"] = RemotePythonValue(axisc) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cross", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cross", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func trapezoid(y: any PythonConvertible, x: (any PythonConvertible)? = nil, dx: Double? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + if let x = x { + __args.append(RemotePythonValue(x)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dx = dx { + __kwargs["dx"] = RemotePythonValue(dx) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "trapezoid", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "trapezoid", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func mean(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "mean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "mean", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func average(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, weights: (any PythonConvertible)? = nil, returned: Bool? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + if let weights = weights { + __args.append(RemotePythonValue(weights)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let returned = returned { + __kwargs["returned"] = RemotePythonValue(returned) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "average", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "average", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func std(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, ddof: Int? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "std", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "std", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func `var`(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, ddof: Int? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "var", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "var", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanmean(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmean", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanstd(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, ddof: Int? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanstd", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanstd", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanvar(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, ddof: Int? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanvar", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanvar", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func min(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "min", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func max(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "max", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func amin(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "amin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "amin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func amax(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "amax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "amax", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanmin(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanmax(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmax", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func ptp(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ptp", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ptp", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func argmin(a: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "argmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "argmin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func argmax(a: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "argmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "argmax", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanargmin(a: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanargmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanargmin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanargmax(a: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanargmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanargmax", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func percentile(a: any PythonConvertible, q: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, overwrite_input: Bool? = nil, method: String? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(q)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let overwrite_input = overwrite_input { + __kwargs["overwrite_input"] = RemotePythonValue(overwrite_input) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "percentile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "percentile", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanpercentile(a: any PythonConvertible, q: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, overwrite_input: Bool? = nil, method: String? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(q)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let overwrite_input = overwrite_input { + __kwargs["overwrite_input"] = RemotePythonValue(overwrite_input) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanpercentile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanpercentile", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func quantile(a: any PythonConvertible, q: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, overwrite_input: Bool? = nil, method: String? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(q)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let overwrite_input = overwrite_input { + __kwargs["overwrite_input"] = RemotePythonValue(overwrite_input) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "quantile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "quantile", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanquantile(a: any PythonConvertible, q: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, overwrite_input: Bool? = nil, method: String? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(q)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let overwrite_input = overwrite_input { + __kwargs["overwrite_input"] = RemotePythonValue(overwrite_input) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanquantile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanquantile", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func median(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, overwrite_input: Bool? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let overwrite_input = overwrite_input { + __kwargs["overwrite_input"] = RemotePythonValue(overwrite_input) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "median", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "median", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nanmedian(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, overwrite_input: Bool? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let overwrite_input = overwrite_input { + __kwargs["overwrite_input"] = RemotePythonValue(overwrite_input) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmedian", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "nanmedian", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func histogram(a: any PythonConvertible, bins: (any PythonConvertible)? = nil, range: (any PythonConvertible)? = nil, density: Bool? = nil, weights: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let bins = bins { + __args.append(RemotePythonValue(bins)) + } + if let range = range { + __args.append(RemotePythonValue(range)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let density = density { + __kwargs["density"] = RemotePythonValue(density) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "histogram", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "histogram", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func histogram2d(x: any PythonConvertible, y: any PythonConvertible, bins: (any PythonConvertible)? = nil, range: (any PythonConvertible)? = nil, density: Bool? = nil, weights: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(y)) + if let bins = bins { + __args.append(RemotePythonValue(bins)) + } + if let range = range { + __args.append(RemotePythonValue(range)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let density = density { + __kwargs["density"] = RemotePythonValue(density) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "histogram2d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "histogram2d", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func histogramdd(sample: any PythonConvertible, bins: (any PythonConvertible)? = nil, range: (any PythonConvertible)? = nil, density: Bool? = nil, weights: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(sample)) + if let bins = bins { + __args.append(RemotePythonValue(bins)) + } + if let range = range { + __args.append(RemotePythonValue(range)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let density = density { + __kwargs["density"] = RemotePythonValue(density) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "histogramdd", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "histogramdd", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func bincount(x: any PythonConvertible, weights: (any PythonConvertible)? = nil, minlength: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + if let weights = weights { + __args.append(RemotePythonValue(weights)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let minlength = minlength { + __kwargs["minlength"] = RemotePythonValue(minlength) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "bincount", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "bincount", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func digitize(x: any PythonConvertible, bins: any PythonConvertible, right: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(bins)) + var __kwargs: [String: RemotePythonValue] = [:] + if let right = right { + __kwargs["right"] = RemotePythonValue(right) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "digitize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "digitize", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func cov(m: any PythonConvertible, y: (any PythonConvertible)? = nil, rowvar: Bool? = nil, bias: Bool? = nil, ddof: Int? = nil, fweights: (any PythonConvertible)? = nil, aweights: (any PythonConvertible)? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(m)) + if let y = y { + __args.append(RemotePythonValue(y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let rowvar = rowvar { + __kwargs["rowvar"] = RemotePythonValue(rowvar) + } + if let bias = bias { + __kwargs["bias"] = RemotePythonValue(bias) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let fweights = fweights { + __kwargs["fweights"] = RemotePythonValue(fweights) + } + if let aweights = aweights { + __kwargs["aweights"] = RemotePythonValue(aweights) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "cov", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "cov", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func corrcoef(x: any PythonConvertible, y: (any PythonConvertible)? = nil, rowvar: Bool? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + if let y = y { + __args.append(RemotePythonValue(y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let rowvar = rowvar { + __kwargs["rowvar"] = RemotePythonValue(rowvar) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "corrcoef", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "corrcoef", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func correlate(a: any PythonConvertible, v: any PythonConvertible, mode: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "correlate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "correlate", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func convolve(a: any PythonConvertible, v: any PythonConvertible, mode: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "convolve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "convolve", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func equal(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "equal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "equal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func not_equal(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "not_equal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "not_equal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func greater(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "greater", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "greater", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func greater_equal(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "greater_equal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "greater_equal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func less(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "less", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "less", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func less_equal(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "less_equal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "less_equal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func maximum(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "maximum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "maximum", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func minimum(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "minimum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "minimum", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fmax(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fmax", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fmin(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fmin", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func clip(a: any PythonConvertible, a_min: any PythonConvertible, a_max: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(a_min)) + __args.append(RemotePythonValue(a_max)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "clip", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "clip", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func allclose(a: any PythonConvertible, b: any PythonConvertible, rtol: Double? = nil, atol: Double? = nil, equal_nan: Bool? = nil, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let rtol = rtol { + __kwargs["rtol"] = RemotePythonValue(rtol) + } + if let atol = atol { + __kwargs["atol"] = RemotePythonValue(atol) + } + if let equal_nan = equal_nan { + __kwargs["equal_nan"] = RemotePythonValue(equal_nan) + } + if let worker { + return try await pool.invokeResult(module: "numpy", function: "allclose", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "allclose", args: __args, kwargs: __kwargs) + } + + public func isclose(a: any PythonConvertible, b: any PythonConvertible, rtol: Double? = nil, atol: Double? = nil, equal_nan: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let rtol = rtol { + __kwargs["rtol"] = RemotePythonValue(rtol) + } + if let atol = atol { + __kwargs["atol"] = RemotePythonValue(atol) + } + if let equal_nan = equal_nan { + __kwargs["equal_nan"] = RemotePythonValue(equal_nan) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isclose", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isclose", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func array_equal(a1: any PythonConvertible, a2: any PythonConvertible, equal_nan: Bool? = nil, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a1)) + __args.append(RemotePythonValue(a2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let equal_nan = equal_nan { + __kwargs["equal_nan"] = RemotePythonValue(equal_nan) + } + if let worker { + return try await pool.invokeResult(module: "numpy", function: "array_equal", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "array_equal", args: __args, kwargs: __kwargs) + } + + public func array_equiv(a1: any PythonConvertible, a2: any PythonConvertible, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a1)) + __args.append(RemotePythonValue(a2)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "numpy", function: "array_equiv", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "array_equiv", args: __args, kwargs: __kwargs) + } + + public func all(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "all", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "all", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func any(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "any", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "any", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func logical_and(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_and", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_and", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logical_or(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_or", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_or", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logical_not(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_not", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_not", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logical_xor(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_xor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "logical_xor", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func bitwise_and(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_and", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_and", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func bitwise_or(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_or", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_or", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func bitwise_xor(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_xor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_xor", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func bitwise_not(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_not", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "bitwise_not", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func invert(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "invert", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "invert", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func left_shift(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "left_shift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "left_shift", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func right_shift(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "right_shift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "right_shift", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isnan(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isnan", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isnan", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isinf(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isinf", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isinf", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isfinite(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isfinite", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isfinite", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isneginf(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isneginf", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isneginf", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isposinf(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isposinf", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isposinf", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isnat(x: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isnat", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isnat", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isreal(x: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isreal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isreal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func iscomplex(x: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "iscomplex", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "iscomplex", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isrealobj(x: any PythonConvertible, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "numpy", function: "isrealobj", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "isrealobj", args: __args, kwargs: __kwargs) + } + + public func iscomplexobj(x: any PythonConvertible, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "numpy", function: "iscomplexobj", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "iscomplexobj", args: __args, kwargs: __kwargs) + } + + public func isscalar(element: any PythonConvertible, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(element)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "numpy", function: "isscalar", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "isscalar", args: __args, kwargs: __kwargs) + } + + public func isfortran(a: any PythonConvertible, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "numpy", function: "isfortran", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "isfortran", args: __args, kwargs: __kwargs) + } + + public func sort(a: any PythonConvertible, axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sort", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func argsort(a: any PythonConvertible, axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "argsort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "argsort", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func lexsort(keys: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(keys)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "lexsort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "lexsort", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func partition(a: any PythonConvertible, kth: any PythonConvertible, axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(kth)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "partition", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "partition", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func argpartition(a: any PythonConvertible, kth: any PythonConvertible, axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(kth)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "argpartition", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "argpartition", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func msort(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "msort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "msort", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sort_complex(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "sort_complex", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "sort_complex", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func count_nonzero(a: any PythonConvertible, axis: (any PythonConvertible)? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "count_nonzero", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "count_nonzero", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func dot(a: any PythonConvertible, b: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "dot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "dot", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func vdot(a: any PythonConvertible, b: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "vdot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "vdot", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inner(a: any PythonConvertible, b: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "inner", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "inner", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func outer(a: any PythonConvertible, b: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "outer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "outer", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func matmul(x1: any PythonConvertible, x2: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x1)) + __args.append(RemotePythonValue(x2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "matmul", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "matmul", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tensordot(a: any PythonConvertible, b: any PythonConvertible, axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "tensordot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "tensordot", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func einsum(_ args: [RemotePythonValue] = [], subscripts: String, out: PoolNDArray? = nil, dtype: String? = nil, order: String? = nil, casting: String? = nil, optimize: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(subscripts)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let casting = casting { + __kwargs["casting"] = RemotePythonValue(casting) + } + if let optimize = optimize { + __kwargs["optimize"] = RemotePythonValue(optimize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "einsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "einsum", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func einsum_path(_ args: [RemotePythonValue] = [], subscripts: String, optimize: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(subscripts)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let optimize = optimize { + __kwargs["optimize"] = RemotePythonValue(optimize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "einsum_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "einsum_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kron(a: any PythonConvertible, b: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "kron", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "kron", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func trace(a: any PythonConvertible, offset: Int? = nil, axis1: Int? = nil, axis2: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let axis1 = axis1 { + __kwargs["axis1"] = RemotePythonValue(axis1) + } + if let axis2 = axis2 { + __kwargs["axis2"] = RemotePythonValue(axis2) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "trace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "trace", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func poly(seq_of_zeros: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(seq_of_zeros)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "poly", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "poly", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func roots(p: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "roots", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "roots", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func polyfit(x: any PythonConvertible, y: any PythonConvertible, deg: Int, rcond: Double? = nil, full: Bool? = nil, w: (any PythonConvertible)? = nil, cov: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(y)) + __args.append(RemotePythonValue(deg)) + var __kwargs: [String: RemotePythonValue] = [:] + if let rcond = rcond { + __kwargs["rcond"] = RemotePythonValue(rcond) + } + if let full = full { + __kwargs["full"] = RemotePythonValue(full) + } + if let w = w { + __kwargs["w"] = RemotePythonValue(w) + } + if let cov = cov { + __kwargs["cov"] = RemotePythonValue(cov) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyfit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyfit", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func polyval(p: any PythonConvertible, x: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyval", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyval", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func polyadd(a1: any PythonConvertible, a2: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a1)) + __args.append(RemotePythonValue(a2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyadd", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyadd", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func polysub(a1: any PythonConvertible, a2: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a1)) + __args.append(RemotePythonValue(a2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polysub", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polysub", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func polymul(a1: any PythonConvertible, a2: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a1)) + __args.append(RemotePythonValue(a2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polymul", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polymul", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func polydiv(u: any PythonConvertible, v: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(u)) + __args.append(RemotePythonValue(v)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polydiv", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polydiv", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func polyint(p: any PythonConvertible, m: Int? = nil, k: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + var __kwargs: [String: RemotePythonValue] = [:] + if let m = m { + __kwargs["m"] = RemotePythonValue(m) + } + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyint", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyint", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func polyder(p: any PythonConvertible, m: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + var __kwargs: [String: RemotePythonValue] = [:] + if let m = m { + __kwargs["m"] = RemotePythonValue(m) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyder", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "polyder", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func intersect1d(ar1: any PythonConvertible, ar2: any PythonConvertible, assume_unique: Bool? = nil, return_indices: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ar1)) + __args.append(RemotePythonValue(ar2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let assume_unique = assume_unique { + __kwargs["assume_unique"] = RemotePythonValue(assume_unique) + } + if let return_indices = return_indices { + __kwargs["return_indices"] = RemotePythonValue(return_indices) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "intersect1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "intersect1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func union1d(ar1: any PythonConvertible, ar2: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ar1)) + __args.append(RemotePythonValue(ar2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "union1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "union1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func setdiff1d(ar1: any PythonConvertible, ar2: any PythonConvertible, assume_unique: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ar1)) + __args.append(RemotePythonValue(ar2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let assume_unique = assume_unique { + __kwargs["assume_unique"] = RemotePythonValue(assume_unique) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "setdiff1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "setdiff1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func setxor1d(ar1: any PythonConvertible, ar2: any PythonConvertible, assume_unique: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ar1)) + __args.append(RemotePythonValue(ar2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let assume_unique = assume_unique { + __kwargs["assume_unique"] = RemotePythonValue(assume_unique) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "setxor1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "setxor1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func in1d(ar1: any PythonConvertible, ar2: any PythonConvertible, assume_unique: Bool? = nil, invert: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ar1)) + __args.append(RemotePythonValue(ar2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let assume_unique = assume_unique { + __kwargs["assume_unique"] = RemotePythonValue(assume_unique) + } + if let invert = invert { + __kwargs["invert"] = RemotePythonValue(invert) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "in1d", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "in1d", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func isin(element: any PythonConvertible, test_elements: any PythonConvertible, assume_unique: Bool? = nil, invert: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(element)) + __args.append(RemotePythonValue(test_elements)) + var __kwargs: [String: RemotePythonValue] = [:] + if let assume_unique = assume_unique { + __kwargs["assume_unique"] = RemotePythonValue(assume_unique) + } + if let invert = invert { + __kwargs["invert"] = RemotePythonValue(invert) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "isin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "isin", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func save(file: any PythonConvertible, arr: any PythonConvertible, allow_pickle: Bool? = nil, fix_imports: Bool? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(file)) + __args.append(RemotePythonValue(arr)) + var __kwargs: [String: RemotePythonValue] = [:] + if let allow_pickle = allow_pickle { + __kwargs["allow_pickle"] = RemotePythonValue(allow_pickle) + } + if let fix_imports = fix_imports { + __kwargs["fix_imports"] = RemotePythonValue(fix_imports) + } + if let worker { + try await pool.invokeVoid(module: "numpy", function: "save", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "save", args: __args, kwargs: __kwargs) + } + + public func savez(_ args: [RemotePythonValue] = [], file: any PythonConvertible, extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(file)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + if let worker { + try await pool.invokeVoid(module: "numpy", function: "savez", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "savez", args: __args, kwargs: __kwargs) + } + + public func savez_compressed(_ args: [RemotePythonValue] = [], file: any PythonConvertible, extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(file)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + if let worker { + try await pool.invokeVoid(module: "numpy", function: "savez_compressed", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "savez_compressed", args: __args, kwargs: __kwargs) + } + + public func load(file: any PythonConvertible, mmap_mode: String? = nil, allow_pickle: Bool? = nil, fix_imports: Bool? = nil, encoding: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(file)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mmap_mode = mmap_mode { + __kwargs["mmap_mode"] = RemotePythonValue(mmap_mode) + } + if let allow_pickle = allow_pickle { + __kwargs["allow_pickle"] = RemotePythonValue(allow_pickle) + } + if let fix_imports = fix_imports { + __kwargs["fix_imports"] = RemotePythonValue(fix_imports) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "load", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "load", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func savetxt(fname: any PythonConvertible, X: any PythonConvertible, fmt: String? = nil, delimiter: String? = nil, newline: String? = nil, header: String? = nil, footer: String? = nil, comments: String? = nil, encoding: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(fname)) + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let fmt = fmt { + __kwargs["fmt"] = RemotePythonValue(fmt) + } + if let delimiter = delimiter { + __kwargs["delimiter"] = RemotePythonValue(delimiter) + } + if let newline = newline { + __kwargs["newline"] = RemotePythonValue(newline) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let footer = footer { + __kwargs["footer"] = RemotePythonValue(footer) + } + if let comments = comments { + __kwargs["comments"] = RemotePythonValue(comments) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let worker { + try await pool.invokeVoid(module: "numpy", function: "savetxt", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "savetxt", args: __args, kwargs: __kwargs) + } + + public func loadtxt(fname: any PythonConvertible, dtype: String? = nil, comments: String? = nil, delimiter: String? = nil, converters: (any PythonConvertible)? = nil, skiprows: Int? = nil, usecols: (any PythonConvertible)? = nil, unpack: Bool? = nil, ndmin: Int? = nil, encoding: String? = nil, max_rows: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(fname)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let comments = comments { + __kwargs["comments"] = RemotePythonValue(comments) + } + if let delimiter = delimiter { + __kwargs["delimiter"] = RemotePythonValue(delimiter) + } + if let converters = converters { + __kwargs["converters"] = RemotePythonValue(converters) + } + if let skiprows = skiprows { + __kwargs["skiprows"] = RemotePythonValue(skiprows) + } + if let usecols = usecols { + __kwargs["usecols"] = RemotePythonValue(usecols) + } + if let unpack = unpack { + __kwargs["unpack"] = RemotePythonValue(unpack) + } + if let ndmin = ndmin { + __kwargs["ndmin"] = RemotePythonValue(ndmin) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let max_rows = max_rows { + __kwargs["max_rows"] = RemotePythonValue(max_rows) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "loadtxt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "loadtxt", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func genfromtxt(fname: any PythonConvertible, dtype: String? = nil, comments: String? = nil, delimiter: (any PythonConvertible)? = nil, skip_header: Int? = nil, skip_footer: Int? = nil, converters: (any PythonConvertible)? = nil, missing_values: (any PythonConvertible)? = nil, filling_values: (any PythonConvertible)? = nil, usecols: (any PythonConvertible)? = nil, names: (any PythonConvertible)? = nil, excludelist: (any PythonConvertible)? = nil, deletechars: String? = nil, replace_space: String? = nil, autostrip: Bool? = nil, case_sensitive: (any PythonConvertible)? = nil, defaultfmt: String? = nil, unpack: Bool? = nil, usemask: Bool? = nil, loose: Bool? = nil, invalid_raise: Bool? = nil, max_rows: Int? = nil, encoding: String? = nil, ndmin: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(fname)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let comments = comments { + __kwargs["comments"] = RemotePythonValue(comments) + } + if let delimiter = delimiter { + __kwargs["delimiter"] = RemotePythonValue(delimiter) + } + if let skip_header = skip_header { + __kwargs["skip_header"] = RemotePythonValue(skip_header) + } + if let skip_footer = skip_footer { + __kwargs["skip_footer"] = RemotePythonValue(skip_footer) + } + if let converters = converters { + __kwargs["converters"] = RemotePythonValue(converters) + } + if let missing_values = missing_values { + __kwargs["missing_values"] = RemotePythonValue(missing_values) + } + if let filling_values = filling_values { + __kwargs["filling_values"] = RemotePythonValue(filling_values) + } + if let usecols = usecols { + __kwargs["usecols"] = RemotePythonValue(usecols) + } + if let names = names { + __kwargs["names"] = RemotePythonValue(names) + } + if let excludelist = excludelist { + __kwargs["excludelist"] = RemotePythonValue(excludelist) + } + if let deletechars = deletechars { + __kwargs["deletechars"] = RemotePythonValue(deletechars) + } + if let replace_space = replace_space { + __kwargs["replace_space"] = RemotePythonValue(replace_space) + } + if let autostrip = autostrip { + __kwargs["autostrip"] = RemotePythonValue(autostrip) + } + if let case_sensitive = case_sensitive { + __kwargs["case_sensitive"] = RemotePythonValue(case_sensitive) + } + if let defaultfmt = defaultfmt { + __kwargs["defaultfmt"] = RemotePythonValue(defaultfmt) + } + if let unpack = unpack { + __kwargs["unpack"] = RemotePythonValue(unpack) + } + if let usemask = usemask { + __kwargs["usemask"] = RemotePythonValue(usemask) + } + if let loose = loose { + __kwargs["loose"] = RemotePythonValue(loose) + } + if let invalid_raise = invalid_raise { + __kwargs["invalid_raise"] = RemotePythonValue(invalid_raise) + } + if let max_rows = max_rows { + __kwargs["max_rows"] = RemotePythonValue(max_rows) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let ndmin = ndmin { + __kwargs["ndmin"] = RemotePythonValue(ndmin) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "genfromtxt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "genfromtxt", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fromfile(file: any PythonConvertible, dtype: String? = nil, count: Int? = nil, sep: String? = nil, offset: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(file)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let count = count { + __kwargs["count"] = RemotePythonValue(count) + } + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "fromfile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "fromfile", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tofile(file: any PythonConvertible, sep: String? = nil, format: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(file)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let format = format { + __kwargs["format"] = RemotePythonValue(format) + } + if let worker { + try await pool.invokeVoid(module: "numpy", function: "tofile", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "tofile", args: __args, kwargs: __kwargs) + } + + public func copy(a: any PythonConvertible, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "copy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "copy", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func copyto(dst: PoolNDArray, src: any PythonConvertible, casting: String? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dst.handle)) + __args.append(RemotePythonValue(src)) + var __kwargs: [String: RemotePythonValue] = [:] + if let casting = casting { + __kwargs["casting"] = RemotePythonValue(casting) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + if let worker { + try await pool.invokeVoid(module: "numpy", function: "copyto", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy", function: "copyto", args: __args, kwargs: __kwargs) + } + + public func asarray(a: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "asarray", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "asarray", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func asanyarray(a: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "asanyarray", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "asanyarray", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ascontiguousarray(a: any PythonConvertible, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ascontiguousarray", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ascontiguousarray", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func asfortranarray(a: any PythonConvertible, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "asfortranarray", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "asfortranarray", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func asarray_chkfinite(a: any PythonConvertible, dtype: String? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "asarray_chkfinite", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "asarray_chkfinite", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func require(a: any PythonConvertible, dtype: String? = nil, requirements: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let requirements = requirements { + __kwargs["requirements"] = RemotePythonValue(requirements) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "require", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "require", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func result_type(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "result_type", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "result_type", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func common_type(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "common_type", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "common_type", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func can_cast(from_: any PythonConvertible, to: any PythonConvertible, casting: String? = nil, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(from_)) + __args.append(RemotePythonValue(to)) + var __kwargs: [String: RemotePythonValue] = [:] + if let casting = casting { + __kwargs["casting"] = RemotePythonValue(casting) + } + if let worker { + return try await pool.invokeResult(module: "numpy", function: "can_cast", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "numpy", function: "can_cast", args: __args, kwargs: __kwargs) + } + + public func promote_types(type1: any PythonConvertible, type2: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(type1)) + __args.append(RemotePythonValue(type2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "promote_types", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "promote_types", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func min_scalar_type(a: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "min_scalar_type", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "min_scalar_type", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func apply_along_axis(_ args: [RemotePythonValue] = [], func1d: any PythonConvertible, axis: Int, arr: any PythonConvertible, extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(func1d)) + __args.append(RemotePythonValue(axis)) + __args.append(RemotePythonValue(arr)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "apply_along_axis", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "apply_along_axis", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func apply_over_axes(`func`: any PythonConvertible, a: any PythonConvertible, axes: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(axes)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "apply_over_axes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "apply_over_axes", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func vectorize(pyfunc: any PythonConvertible, otypes: (any PythonConvertible)? = nil, doc: String? = nil, excluded: (any PythonConvertible)? = nil, cache: Bool? = nil, signature: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(pyfunc)) + if let otypes = otypes { + __args.append(RemotePythonValue(otypes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let doc = doc { + __kwargs["doc"] = RemotePythonValue(doc) + } + if let excluded = excluded { + __kwargs["excluded"] = RemotePythonValue(excluded) + } + if let cache = cache { + __kwargs["cache"] = RemotePythonValue(cache) + } + if let signature = signature { + __kwargs["signature"] = RemotePythonValue(signature) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "vectorize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "vectorize", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func piecewise(_ args: [RemotePythonValue] = [], x: any PythonConvertible, condlist: any PythonConvertible, funclist: any PythonConvertible, extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(condlist)) + __args.append(RemotePythonValue(funclist)) + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "piecewise", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "piecewise", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func interp(x: any PythonConvertible, xp: any PythonConvertible, fp: any PythonConvertible, left: Double? = nil, right: Double? = nil, period: Double? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(xp)) + __args.append(RemotePythonValue(fp)) + var __kwargs: [String: RemotePythonValue] = [:] + if let left = left { + __kwargs["left"] = RemotePythonValue(left) + } + if let right = right { + __kwargs["right"] = RemotePythonValue(right) + } + if let period = period { + __kwargs["period"] = RemotePythonValue(period) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "interp", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "interp", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ndarray(shape: any PythonConvertible, dtype: String? = nil, buffer: (any PythonConvertible)? = nil, offset: Int? = nil, strides: (any PythonConvertible)? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let buffer = buffer { + __kwargs["buffer"] = RemotePythonValue(buffer) + } + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let strides = strides { + __kwargs["strides"] = RemotePythonValue(strides) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy", function: "ndarray", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy", function: "ndarray", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } +} + + +public struct PoolNDArray: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func reshape(shape: any PythonConvertible, order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "reshape", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "reshape", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func resize(new_shape: any PythonConvertible, refcheck: Bool? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(new_shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let refcheck = refcheck { + __kwargs["refcheck"] = RemotePythonValue(refcheck) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "resize", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "resize", args: __args, kwargs: __kwargs) + } + + public func flatten(order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "flatten", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "flatten", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ravel(order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ravel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ravel", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func transpose(axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axes = axes { + __kwargs["axes"] = RemotePythonValue(axes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transpose", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transpose", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func swapaxes(axis1: Int, axis2: Int, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(axis1)) + __args.append(RemotePythonValue(axis2)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "swapaxes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "swapaxes", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func squeeze(axis: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "squeeze", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "squeeze", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func astype(dtype: String, order: String? = nil, casting: String? = nil, subok: Bool? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let casting = casting { + __kwargs["casting"] = RemotePythonValue(casting) + } + if let subok = subok { + __kwargs["subok"] = RemotePythonValue(subok) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func copy(order: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func view(dtype: String? = nil, type: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let type = type { + __kwargs["type"] = RemotePythonValue(type) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "view", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "view", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fill(value: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(value)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + _ = try await pool.methodOwned(handle: self, name: "fill", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "fill", args: __args, kwargs: __kwargs) + } + + public func tolist(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tolist", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tolist", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func tobytes(order: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tobytes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tobytes", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func tofile(fid: any PythonConvertible, sep: String? = nil, format: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(fid)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let format = format { + __kwargs["format"] = RemotePythonValue(format) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "tofile", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "tofile", args: __args, kwargs: __kwargs) + } + + public func tostring(order: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tostring", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tostring", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func item(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "item", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "item", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func itemset(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + _ = try await pool.methodOwned(handle: self, name: "itemset", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "itemset", args: __args, kwargs: __kwargs) + } + + public func byteswap(inplace: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "byteswap", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "byteswap", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func getfield(dtype: String, offset: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "getfield", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "getfield", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func setfield(val: any PythonConvertible, dtype: String, offset: Int? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(val)) + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "setfield", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "setfield", args: __args, kwargs: __kwargs) + } + + public func setflags(write: Bool? = nil, align: Bool? = nil, uic: Bool? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let write = write { + __kwargs["write"] = RemotePythonValue(write) + } + if let align = align { + __kwargs["align"] = RemotePythonValue(align) + } + if let uic = uic { + __kwargs["uic"] = RemotePythonValue(uic) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "setflags", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "setflags", args: __args, kwargs: __kwargs) + } + + public func take(indices: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, mode: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(indices)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "take", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "take", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func put(indices: any PythonConvertible, values: any PythonConvertible, mode: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(indices)) + __args.append(RemotePythonValue(values)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "put", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "put", args: __args, kwargs: __kwargs) + } + + public func `repeat`(repeats: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(repeats)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "repeat", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "repeat", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func choose(choices: any PythonConvertible, out: PoolNDArray? = nil, mode: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(choices)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "choose", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "choose", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sort(axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "sort", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "sort", args: __args, kwargs: __kwargs) + } + + public func argsort(axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "argsort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "argsort", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func partition(kth: any PythonConvertible, axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(kth)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "partition", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "partition", args: __args, kwargs: __kwargs) + } + + public func argpartition(kth: any PythonConvertible, axis: Int? = nil, kind: String? = nil, order: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(kth)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "argpartition", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "argpartition", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func searchsorted(v: any PythonConvertible, side: String? = nil, sorter: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(v)) + var __kwargs: [String: RemotePythonValue] = [:] + if let side = side { + __kwargs["side"] = RemotePythonValue(side) + } + if let sorter = sorter { + __kwargs["sorter"] = RemotePythonValue(sorter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "searchsorted", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "searchsorted", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func nonzero(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nonzero", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nonzero", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func compress(condition: any PythonConvertible, axis: Int? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(condition)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "compress", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "compress", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func diagonal(offset: Int? = nil, axis1: Int? = nil, axis2: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let axis1 = axis1 { + __kwargs["axis1"] = RemotePythonValue(axis1) + } + if let axis2 = axis2 { + __kwargs["axis2"] = RemotePythonValue(axis2) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "diagonal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "diagonal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sum(axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func prod(axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cumsum(axis: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func cumprod(axis: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func mean(axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func std(axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, ddof: Int? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func `var`(axis: (any PythonConvertible)? = nil, dtype: String? = nil, out: PoolNDArray? = nil, ddof: Int? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func min(axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func max(axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, initial: (any PythonConvertible)? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let initial = initial { + __kwargs["initial"] = RemotePythonValue(initial) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func ptp(axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ptp", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ptp", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func argmin(axis: Int? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "argmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "argmin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func argmax(axis: Int? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "argmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "argmax", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func clip(a_min: any PythonConvertible, a_max: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a_min)) + __args.append(RemotePythonValue(a_max)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "clip", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "clip", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func round(decimals: Int? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let decimals = decimals { + __kwargs["decimals"] = RemotePythonValue(decimals) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "round", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "round", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func trace(offset: Int? = nil, axis1: Int? = nil, axis2: Int? = nil, dtype: String? = nil, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let axis1 = axis1 { + __kwargs["axis1"] = RemotePythonValue(axis1) + } + if let axis2 = axis2 { + __kwargs["axis2"] = RemotePythonValue(axis2) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "trace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "trace", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func all(axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "all", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "all", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func any(axis: (any PythonConvertible)? = nil, out: PoolNDArray? = nil, keepdims: Bool? = nil, where where_: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + if let where_ = where_ { + __kwargs["where"] = RemotePythonValue(where_) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "any", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "any", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func dot(b: any PythonConvertible, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "dot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "dot", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func conj(worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "conj", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "conj", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func conjugate(worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "conjugate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "conjugate", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } +} + +// MARK: - NumPy.Fft + +extension PoolNumPy { + /// Pool-bound `numpy.fft` namespace. Calls run on this pool. + public var fft: PoolNumPyFft { PoolNumPyFft(pool: pool) } +} + +public struct PoolNumPyFft: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func fft(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fft", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fft", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ifft(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifft", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifft", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fft2(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fft2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fft2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ifft2(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifft2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifft2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fftn(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fftn", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fftn", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ifftn(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifftn", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifftn", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rfft(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfft", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfft", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func irfft(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "irfft", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "irfft", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rfft2(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfft2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfft2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func irfft2(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "irfft2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "irfft2", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rfftn(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfftn", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfftn", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func irfftn(a: any PythonConvertible, s: (any PythonConvertible)? = nil, axes: (any PythonConvertible)? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let s = s { + __args.append(RemotePythonValue(s)) + } + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "irfftn", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "irfftn", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func hfft(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "hfft", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "hfft", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ihfft(a: any PythonConvertible, n: Int? = nil, axis: Int? = nil, norm: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ihfft", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ihfft", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fftfreq(n: Int, d: Double? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + var __kwargs: [String: RemotePythonValue] = [:] + if let d = d { + __kwargs["d"] = RemotePythonValue(d) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fftfreq", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fftfreq", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rfftfreq(n: Int, d: Double? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + var __kwargs: [String: RemotePythonValue] = [:] + if let d = d { + __kwargs["d"] = RemotePythonValue(d) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfftfreq", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "rfftfreq", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func fftshift(x: any PythonConvertible, axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fftshift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "fftshift", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ifftshift(x: any PythonConvertible, axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifftshift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.fft", function: "ifftshift", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } +} + +// MARK: - NumPy.Linalg + +extension PoolNumPy { + /// Pool-bound `numpy.linalg` namespace. Calls run on this pool. + public var linalg: PoolNumPyLinalg { PoolNumPyLinalg(pool: pool) } +} + +public struct PoolNumPyLinalg: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func inv(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "inv", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "inv", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func pinv(a: any PythonConvertible, rcond: Double? = nil, hermitian: Bool? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let rcond = rcond { + __kwargs["rcond"] = RemotePythonValue(rcond) + } + if let hermitian = hermitian { + __kwargs["hermitian"] = RemotePythonValue(hermitian) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "pinv", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "pinv", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func det(a: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "det", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "det", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func slogdet(a: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "slogdet", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "slogdet", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func matrix_rank(M: any PythonConvertible, tol: Double? = nil, hermitian: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(M)) + var __kwargs: [String: RemotePythonValue] = [:] + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let hermitian = hermitian { + __kwargs["hermitian"] = RemotePythonValue(hermitian) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "matrix_rank", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "matrix_rank", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func solve(a: any PythonConvertible, b: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "solve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "solve", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func lstsq(a: any PythonConvertible, b: any PythonConvertible, rcond: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + var __kwargs: [String: RemotePythonValue] = [:] + if let rcond = rcond { + __kwargs["rcond"] = RemotePythonValue(rcond) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "lstsq", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "lstsq", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func eig(a: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eig", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eig", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func eigh(a: any PythonConvertible, UPLO: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let UPLO = UPLO { + __kwargs["UPLO"] = RemotePythonValue(UPLO) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eigh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eigh", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func eigvals(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eigvals", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eigvals", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func eigvalsh(a: any PythonConvertible, UPLO: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let UPLO = UPLO { + __kwargs["UPLO"] = RemotePythonValue(UPLO) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eigvalsh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "eigvalsh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func svd(a: any PythonConvertible, full_matrices: Bool? = nil, compute_uv: Bool? = nil, hermitian: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let full_matrices = full_matrices { + __kwargs["full_matrices"] = RemotePythonValue(full_matrices) + } + if let compute_uv = compute_uv { + __kwargs["compute_uv"] = RemotePythonValue(compute_uv) + } + if let hermitian = hermitian { + __kwargs["hermitian"] = RemotePythonValue(hermitian) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "svd", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "svd", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func svdvals(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "svdvals", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "svdvals", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func norm(x: any PythonConvertible, ord: (any PythonConvertible)? = nil, axis: (any PythonConvertible)? = nil, keepdims: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + if let ord = ord { + __args.append(RemotePythonValue(ord)) + } + if let axis = axis { + __args.append(RemotePythonValue(axis)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let keepdims = keepdims { + __kwargs["keepdims"] = RemotePythonValue(keepdims) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "norm", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "norm", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cond(x: any PythonConvertible, p: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + if let p = p { + __args.append(RemotePythonValue(p)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "cond", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "cond", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func qr(a: any PythonConvertible, mode: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "qr", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "qr", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cholesky(a: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "cholesky", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "cholesky", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func matrix_power(a: any PythonConvertible, n: Int, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(n)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "matrix_power", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "matrix_power", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func multi_dot(arrays: PyHandle, out: PoolNDArray? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arrays)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out = out { + __kwargs["out"] = RemotePythonValue(out.handle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "multi_dot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "multi_dot", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tensorinv(a: any PythonConvertible, ind: Int? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + var __kwargs: [String: RemotePythonValue] = [:] + if let ind = ind { + __kwargs["ind"] = RemotePythonValue(ind) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "tensorinv", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "tensorinv", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func tensorsolve(a: any PythonConvertible, b: any PythonConvertible, axes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + if let axes = axes { + __args.append(RemotePythonValue(axes)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "tensorsolve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.linalg", function: "tensorsolve", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } +} + +// MARK: - NumPy.Random + +extension PoolNumPy { + /// Pool-bound `numpy.random` namespace. Calls run on this pool. + public var random: PoolNumPyRandom { PoolNumPyRandom(pool: pool) } +} + +public struct PoolNumPyRandom: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func seed(seed: Int? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let seed = seed { + __kwargs["seed"] = RemotePythonValue(seed) + } + if let worker { + try await pool.invokeVoid(module: "numpy.random", function: "seed", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy.random", function: "seed", args: __args, kwargs: __kwargs) + } + + public func rand(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "rand", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "rand", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func randn(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "randn", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "randn", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func randint(low: Int, high: Int? = nil, size: (any PythonConvertible)? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(low)) + var __kwargs: [String: RemotePythonValue] = [:] + if let high = high { + __kwargs["high"] = RemotePythonValue(high) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "randint", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "randint", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func random(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "random", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "random", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func random_sample(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "random_sample", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "random_sample", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func ranf(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "ranf", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "ranf", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func sample(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "sample", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "sample", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func choice(a: any PythonConvertible, size: (any PythonConvertible)? = nil, replace: Bool? = nil, p: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let replace = replace { + __kwargs["replace"] = RemotePythonValue(replace) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "choice", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "choice", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func shuffle(x: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + try await pool.invokeVoid(module: "numpy.random", function: "shuffle", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "numpy.random", function: "shuffle", args: __args, kwargs: __kwargs) + } + + public func permutation(x: any PythonConvertible, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "permutation", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "permutation", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func uniform(low: Double? = nil, high: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let low = low { + __kwargs["low"] = RemotePythonValue(low) + } + if let high = high { + __kwargs["high"] = RemotePythonValue(high) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "uniform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "uniform", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func normal(loc: Double? = nil, scale: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loc = loc { + __kwargs["loc"] = RemotePythonValue(loc) + } + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "normal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "normal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func standard_normal(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_normal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_normal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func binomial(n: Int, p: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + __args.append(RemotePythonValue(p)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "binomial", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "binomial", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func poisson(lam: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let lam = lam { + __kwargs["lam"] = RemotePythonValue(lam) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "poisson", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "poisson", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func exponential(scale: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "exponential", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "exponential", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func gamma(shape: Double, scale: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + var __kwargs: [String: RemotePythonValue] = [:] + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "gamma", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "gamma", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func beta(a: Double, b: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + __args.append(RemotePythonValue(b)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "beta", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "beta", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func chisquare(df: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(df)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "chisquare", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "chisquare", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func f(dfnum: Double, dfden: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dfnum)) + __args.append(RemotePythonValue(dfden)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "f", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "f", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func geometric(p: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "geometric", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "geometric", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func hypergeometric(ngood: Int, nbad: Int, nsample: Int, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(ngood)) + __args.append(RemotePythonValue(nbad)) + __args.append(RemotePythonValue(nsample)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "hypergeometric", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "hypergeometric", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func laplace(loc: Double? = nil, scale: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loc = loc { + __kwargs["loc"] = RemotePythonValue(loc) + } + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "laplace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "laplace", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func logistic(loc: Double? = nil, scale: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loc = loc { + __kwargs["loc"] = RemotePythonValue(loc) + } + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "logistic", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "logistic", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func lognormal(mean: Double? = nil, sigma: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let mean = mean { + __kwargs["mean"] = RemotePythonValue(mean) + } + if let sigma = sigma { + __kwargs["sigma"] = RemotePythonValue(sigma) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "lognormal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "lognormal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func multinomial(n: Int, pvals: any PythonConvertible, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + __args.append(RemotePythonValue(pvals)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "multinomial", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "multinomial", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func multivariate_normal(mean: any PythonConvertible, cov: any PythonConvertible, size: (any PythonConvertible)? = nil, check_valid: String? = nil, tol: Double? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(mean)) + __args.append(RemotePythonValue(cov)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let check_valid = check_valid { + __kwargs["check_valid"] = RemotePythonValue(check_valid) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "multivariate_normal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "multivariate_normal", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func negative_binomial(n: Double, p: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + __args.append(RemotePythonValue(p)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "negative_binomial", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "negative_binomial", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func pareto(a: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "pareto", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "pareto", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func power(a: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "power", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "power", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func rayleigh(scale: Double? = nil, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let size = size { + __kwargs["size"] = RemotePythonValue(size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "rayleigh", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "rayleigh", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func standard_cauchy(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_cauchy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_cauchy", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func standard_exponential(size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_exponential", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_exponential", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func standard_gamma(shape: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_gamma", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_gamma", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func standard_t(df: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(df)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_t", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "standard_t", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func triangular(left: Double, mode: Double, right: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left)) + __args.append(RemotePythonValue(mode)) + __args.append(RemotePythonValue(right)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "triangular", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "triangular", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func vonmises(mu: Double, kappa: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(mu)) + __args.append(RemotePythonValue(kappa)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "vonmises", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "vonmises", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func wald(mean: Double, scale: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(mean)) + __args.append(RemotePythonValue(scale)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "wald", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "wald", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func weibull(a: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "weibull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "weibull", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } + + public func zipf(a: Double, size: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolNDArray { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(a)) + if let size = size { + __args.append(RemotePythonValue(size)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "zipf", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "numpy.random", function: "zipf", args: __args, kwargs: __kwargs) + } + return PoolNDArray(pool: pool, owned: __owned) + } +} + diff --git a/Examples/Studio/Sources/SwiftPythonKit/PoolPandas.swift b/Examples/Studio/Sources/SwiftPythonKit/PoolPandas.swift new file mode 100644 index 0000000..6cd4f83 --- /dev/null +++ b/Examples/Studio/Sources/SwiftPythonKit/PoolPandas.swift @@ -0,0 +1,8174 @@ +// Generated by swift-python-gen +// Do not edit manually + +import SwiftPythonRuntime +import Foundation + +// MARK: - Pandas + +extension PythonProcessPool { + /// Pool-bound `pandas` namespace. Calls run on this pool. + public var pandas: PoolPandas { PoolPandas(pool: self) } +} + +public struct PoolPandas: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func read_csv(filepath_or_buffer: String, sep: String? = nil, delimiter: String? = nil, header: (any PythonConvertible)? = nil, names: [String]? = nil, index_col: (any PythonConvertible)? = nil, usecols: (any PythonConvertible)? = nil, dtype: (any PythonConvertible)? = nil, engine: String? = nil, converters: (any PythonConvertible)? = nil, true_values: [String]? = nil, false_values: [String]? = nil, skipinitialspace: Bool? = nil, skiprows: (any PythonConvertible)? = nil, skipfooter: Int? = nil, nrows: Int? = nil, na_values: (any PythonConvertible)? = nil, keep_default_na: Bool? = nil, na_filter: Bool? = nil, verbose: Bool? = nil, skip_blank_lines: Bool? = nil, parse_dates: (any PythonConvertible)? = nil, infer_datetime_format: Bool? = nil, keep_date_col: Bool? = nil, date_parser: (any PythonConvertible)? = nil, dayfirst: Bool? = nil, cache_dates: Bool? = nil, iterator: Bool? = nil, chunksize: Int? = nil, compression: String? = nil, thousands: String? = nil, decimal: String? = nil, lineterminator: String? = nil, quotechar: String? = nil, quoting: Int? = nil, doublequote: Bool? = nil, escapechar: String? = nil, comment: String? = nil, encoding: String? = nil, encoding_errors: String? = nil, dialect: String? = nil, on_bad_lines: String? = nil, delim_whitespace: Bool? = nil, low_memory: Bool? = nil, memory_map: Bool? = nil, float_precision: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(filepath_or_buffer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let delimiter = delimiter { + __kwargs["delimiter"] = RemotePythonValue(delimiter) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let names = names { + __kwargs["names"] = RemotePythonValue(names) + } + if let index_col = index_col { + __kwargs["index_col"] = RemotePythonValue(index_col) + } + if let usecols = usecols { + __kwargs["usecols"] = RemotePythonValue(usecols) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let engine = engine { + __kwargs["engine"] = RemotePythonValue(engine) + } + if let converters = converters { + __kwargs["converters"] = RemotePythonValue(converters) + } + if let true_values = true_values { + __kwargs["true_values"] = RemotePythonValue(true_values) + } + if let false_values = false_values { + __kwargs["false_values"] = RemotePythonValue(false_values) + } + if let skipinitialspace = skipinitialspace { + __kwargs["skipinitialspace"] = RemotePythonValue(skipinitialspace) + } + if let skiprows = skiprows { + __kwargs["skiprows"] = RemotePythonValue(skiprows) + } + if let skipfooter = skipfooter { + __kwargs["skipfooter"] = RemotePythonValue(skipfooter) + } + if let nrows = nrows { + __kwargs["nrows"] = RemotePythonValue(nrows) + } + if let na_values = na_values { + __kwargs["na_values"] = RemotePythonValue(na_values) + } + if let keep_default_na = keep_default_na { + __kwargs["keep_default_na"] = RemotePythonValue(keep_default_na) + } + if let na_filter = na_filter { + __kwargs["na_filter"] = RemotePythonValue(na_filter) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let skip_blank_lines = skip_blank_lines { + __kwargs["skip_blank_lines"] = RemotePythonValue(skip_blank_lines) + } + if let parse_dates = parse_dates { + __kwargs["parse_dates"] = RemotePythonValue(parse_dates) + } + if let infer_datetime_format = infer_datetime_format { + __kwargs["infer_datetime_format"] = RemotePythonValue(infer_datetime_format) + } + if let keep_date_col = keep_date_col { + __kwargs["keep_date_col"] = RemotePythonValue(keep_date_col) + } + if let date_parser = date_parser { + __kwargs["date_parser"] = RemotePythonValue(date_parser) + } + if let dayfirst = dayfirst { + __kwargs["dayfirst"] = RemotePythonValue(dayfirst) + } + if let cache_dates = cache_dates { + __kwargs["cache_dates"] = RemotePythonValue(cache_dates) + } + if let iterator = iterator { + __kwargs["iterator"] = RemotePythonValue(iterator) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let thousands = thousands { + __kwargs["thousands"] = RemotePythonValue(thousands) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let lineterminator = lineterminator { + __kwargs["lineterminator"] = RemotePythonValue(lineterminator) + } + if let quotechar = quotechar { + __kwargs["quotechar"] = RemotePythonValue(quotechar) + } + if let quoting = quoting { + __kwargs["quoting"] = RemotePythonValue(quoting) + } + if let doublequote = doublequote { + __kwargs["doublequote"] = RemotePythonValue(doublequote) + } + if let escapechar = escapechar { + __kwargs["escapechar"] = RemotePythonValue(escapechar) + } + if let comment = comment { + __kwargs["comment"] = RemotePythonValue(comment) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let encoding_errors = encoding_errors { + __kwargs["encoding_errors"] = RemotePythonValue(encoding_errors) + } + if let dialect = dialect { + __kwargs["dialect"] = RemotePythonValue(dialect) + } + if let on_bad_lines = on_bad_lines { + __kwargs["on_bad_lines"] = RemotePythonValue(on_bad_lines) + } + if let delim_whitespace = delim_whitespace { + __kwargs["delim_whitespace"] = RemotePythonValue(delim_whitespace) + } + if let low_memory = low_memory { + __kwargs["low_memory"] = RemotePythonValue(low_memory) + } + if let memory_map = memory_map { + __kwargs["memory_map"] = RemotePythonValue(memory_map) + } + if let float_precision = float_precision { + __kwargs["float_precision"] = RemotePythonValue(float_precision) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_csv", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_csv", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_excel(io: any PythonConvertible, sheet_name: (any PythonConvertible)? = nil, header: (any PythonConvertible)? = nil, names: [String]? = nil, index_col: (any PythonConvertible)? = nil, usecols: (any PythonConvertible)? = nil, dtype: (any PythonConvertible)? = nil, engine: String? = nil, converters: (any PythonConvertible)? = nil, true_values: [String]? = nil, false_values: [String]? = nil, skiprows: (any PythonConvertible)? = nil, nrows: Int? = nil, na_values: (any PythonConvertible)? = nil, keep_default_na: Bool? = nil, na_filter: Bool? = nil, verbose: Bool? = nil, parse_dates: (any PythonConvertible)? = nil, date_parser: (any PythonConvertible)? = nil, thousands: String? = nil, decimal: String? = nil, comment: String? = nil, skipfooter: Int? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(io)) + if let sheet_name = sheet_name { + __args.append(RemotePythonValue(sheet_name)) + } + if let header = header { + __args.append(RemotePythonValue(header)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let names = names { + __kwargs["names"] = RemotePythonValue(names) + } + if let index_col = index_col { + __kwargs["index_col"] = RemotePythonValue(index_col) + } + if let usecols = usecols { + __kwargs["usecols"] = RemotePythonValue(usecols) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let engine = engine { + __kwargs["engine"] = RemotePythonValue(engine) + } + if let converters = converters { + __kwargs["converters"] = RemotePythonValue(converters) + } + if let true_values = true_values { + __kwargs["true_values"] = RemotePythonValue(true_values) + } + if let false_values = false_values { + __kwargs["false_values"] = RemotePythonValue(false_values) + } + if let skiprows = skiprows { + __kwargs["skiprows"] = RemotePythonValue(skiprows) + } + if let nrows = nrows { + __kwargs["nrows"] = RemotePythonValue(nrows) + } + if let na_values = na_values { + __kwargs["na_values"] = RemotePythonValue(na_values) + } + if let keep_default_na = keep_default_na { + __kwargs["keep_default_na"] = RemotePythonValue(keep_default_na) + } + if let na_filter = na_filter { + __kwargs["na_filter"] = RemotePythonValue(na_filter) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let parse_dates = parse_dates { + __kwargs["parse_dates"] = RemotePythonValue(parse_dates) + } + if let date_parser = date_parser { + __kwargs["date_parser"] = RemotePythonValue(date_parser) + } + if let thousands = thousands { + __kwargs["thousands"] = RemotePythonValue(thousands) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let comment = comment { + __kwargs["comment"] = RemotePythonValue(comment) + } + if let skipfooter = skipfooter { + __kwargs["skipfooter"] = RemotePythonValue(skipfooter) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_excel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_excel", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_json(path_or_buf: String? = nil, orient: String? = nil, typ: String? = nil, dtype: (any PythonConvertible)? = nil, convert_axes: Bool? = nil, convert_dates: Bool? = nil, keep_default_dates: Bool? = nil, precise_float: Bool? = nil, date_unit: String? = nil, encoding: String? = nil, encoding_errors: String? = nil, lines: Bool? = nil, chunksize: Int? = nil, compression: String? = nil, nrows: Int? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let path_or_buf = path_or_buf { + __kwargs["path_or_buf"] = RemotePythonValue(path_or_buf) + } + if let orient = orient { + __kwargs["orient"] = RemotePythonValue(orient) + } + if let typ = typ { + __kwargs["typ"] = RemotePythonValue(typ) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let convert_axes = convert_axes { + __kwargs["convert_axes"] = RemotePythonValue(convert_axes) + } + if let convert_dates = convert_dates { + __kwargs["convert_dates"] = RemotePythonValue(convert_dates) + } + if let keep_default_dates = keep_default_dates { + __kwargs["keep_default_dates"] = RemotePythonValue(keep_default_dates) + } + if let precise_float = precise_float { + __kwargs["precise_float"] = RemotePythonValue(precise_float) + } + if let date_unit = date_unit { + __kwargs["date_unit"] = RemotePythonValue(date_unit) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let encoding_errors = encoding_errors { + __kwargs["encoding_errors"] = RemotePythonValue(encoding_errors) + } + if let lines = lines { + __kwargs["lines"] = RemotePythonValue(lines) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let nrows = nrows { + __kwargs["nrows"] = RemotePythonValue(nrows) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_json", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_json", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_parquet(path: String, engine: String? = nil, columns: [String]? = nil, storage_options: (any PythonConvertible)? = nil, use_nullable_dtypes: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(path)) + var __kwargs: [String: RemotePythonValue] = [:] + if let engine = engine { + __kwargs["engine"] = RemotePythonValue(engine) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + if let use_nullable_dtypes = use_nullable_dtypes { + __kwargs["use_nullable_dtypes"] = RemotePythonValue(use_nullable_dtypes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_parquet", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_parquet", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_sql(sql: String, con: any PythonConvertible, index_col: (any PythonConvertible)? = nil, coerce_float: Bool? = nil, params: (any PythonConvertible)? = nil, parse_dates: (any PythonConvertible)? = nil, columns: [String]? = nil, chunksize: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(sql)) + __args.append(RemotePythonValue(con)) + if let index_col = index_col { + __args.append(RemotePythonValue(index_col)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let coerce_float = coerce_float { + __kwargs["coerce_float"] = RemotePythonValue(coerce_float) + } + if let params = params { + __kwargs["params"] = RemotePythonValue(params) + } + if let parse_dates = parse_dates { + __kwargs["parse_dates"] = RemotePythonValue(parse_dates) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sql", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sql", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_sql_query(sql: String, con: any PythonConvertible, index_col: (any PythonConvertible)? = nil, coerce_float: Bool? = nil, params: (any PythonConvertible)? = nil, parse_dates: (any PythonConvertible)? = nil, chunksize: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(sql)) + __args.append(RemotePythonValue(con)) + if let index_col = index_col { + __args.append(RemotePythonValue(index_col)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let coerce_float = coerce_float { + __kwargs["coerce_float"] = RemotePythonValue(coerce_float) + } + if let params = params { + __kwargs["params"] = RemotePythonValue(params) + } + if let parse_dates = parse_dates { + __kwargs["parse_dates"] = RemotePythonValue(parse_dates) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sql_query", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sql_query", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_sql_table(table_name: String, con: any PythonConvertible, schema: String? = nil, index_col: (any PythonConvertible)? = nil, coerce_float: Bool? = nil, parse_dates: (any PythonConvertible)? = nil, columns: [String]? = nil, chunksize: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(table_name)) + __args.append(RemotePythonValue(con)) + var __kwargs: [String: RemotePythonValue] = [:] + if let schema = schema { + __kwargs["schema"] = RemotePythonValue(schema) + } + if let index_col = index_col { + __kwargs["index_col"] = RemotePythonValue(index_col) + } + if let coerce_float = coerce_float { + __kwargs["coerce_float"] = RemotePythonValue(coerce_float) + } + if let parse_dates = parse_dates { + __kwargs["parse_dates"] = RemotePythonValue(parse_dates) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sql_table", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sql_table", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_html(io: any PythonConvertible, match: String? = nil, flavor: String? = nil, header: (any PythonConvertible)? = nil, index_col: (any PythonConvertible)? = nil, skiprows: (any PythonConvertible)? = nil, attrs: (any PythonConvertible)? = nil, parse_dates: Bool? = nil, thousands: String? = nil, encoding: String? = nil, decimal: String? = nil, converters: (any PythonConvertible)? = nil, na_values: (any PythonConvertible)? = nil, keep_default_na: Bool? = nil, displayed_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(io)) + var __kwargs: [String: RemotePythonValue] = [:] + if let match = match { + __kwargs["match"] = RemotePythonValue(match) + } + if let flavor = flavor { + __kwargs["flavor"] = RemotePythonValue(flavor) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index_col = index_col { + __kwargs["index_col"] = RemotePythonValue(index_col) + } + if let skiprows = skiprows { + __kwargs["skiprows"] = RemotePythonValue(skiprows) + } + if let attrs = attrs { + __kwargs["attrs"] = RemotePythonValue(attrs) + } + if let parse_dates = parse_dates { + __kwargs["parse_dates"] = RemotePythonValue(parse_dates) + } + if let thousands = thousands { + __kwargs["thousands"] = RemotePythonValue(thousands) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let converters = converters { + __kwargs["converters"] = RemotePythonValue(converters) + } + if let na_values = na_values { + __kwargs["na_values"] = RemotePythonValue(na_values) + } + if let keep_default_na = keep_default_na { + __kwargs["keep_default_na"] = RemotePythonValue(keep_default_na) + } + if let displayed_only = displayed_only { + __kwargs["displayed_only"] = RemotePythonValue(displayed_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_html", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_html", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func read_pickle(filepath_or_buffer: String, compression: String? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(filepath_or_buffer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_pickle", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_pickle", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func read_clipboard(sep: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_clipboard", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_clipboard", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_fwf(filepath_or_buffer: String, colspecs: (any PythonConvertible)? = nil, widths: [Int]? = nil, infer_nrows: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(filepath_or_buffer)) + if let colspecs = colspecs { + __args.append(RemotePythonValue(colspecs)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let widths = widths { + __kwargs["widths"] = RemotePythonValue(widths) + } + if let infer_nrows = infer_nrows { + __kwargs["infer_nrows"] = RemotePythonValue(infer_nrows) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_fwf", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_fwf", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_feather(path: String, columns: [String]? = nil, use_threads: Bool? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(path)) + var __kwargs: [String: RemotePythonValue] = [:] + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let use_threads = use_threads { + __kwargs["use_threads"] = RemotePythonValue(use_threads) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_feather", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_feather", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_orc(path: String, columns: [String]? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(path)) + var __kwargs: [String: RemotePythonValue] = [:] + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_orc", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_orc", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_stata(filepath_or_buffer: String, convert_dates: Bool? = nil, convert_categoricals: Bool? = nil, index_col: String? = nil, convert_missing: Bool? = nil, preserve_dtypes: Bool? = nil, columns: [String]? = nil, order_categoricals: Bool? = nil, chunksize: Int? = nil, iterator: Bool? = nil, compression: String? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(filepath_or_buffer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let convert_dates = convert_dates { + __kwargs["convert_dates"] = RemotePythonValue(convert_dates) + } + if let convert_categoricals = convert_categoricals { + __kwargs["convert_categoricals"] = RemotePythonValue(convert_categoricals) + } + if let index_col = index_col { + __kwargs["index_col"] = RemotePythonValue(index_col) + } + if let convert_missing = convert_missing { + __kwargs["convert_missing"] = RemotePythonValue(convert_missing) + } + if let preserve_dtypes = preserve_dtypes { + __kwargs["preserve_dtypes"] = RemotePythonValue(preserve_dtypes) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let order_categoricals = order_categoricals { + __kwargs["order_categoricals"] = RemotePythonValue(order_categoricals) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let iterator = iterator { + __kwargs["iterator"] = RemotePythonValue(iterator) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_stata", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_stata", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_sas(filepath_or_buffer: String, format: String? = nil, index: String? = nil, encoding: String? = nil, chunksize: Int? = nil, iterator: Bool? = nil, compression: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(filepath_or_buffer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let format = format { + __kwargs["format"] = RemotePythonValue(format) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let iterator = iterator { + __kwargs["iterator"] = RemotePythonValue(iterator) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sas", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_sas", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_spss(path: String, usecols: [String]? = nil, convert_categoricals: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(path)) + var __kwargs: [String: RemotePythonValue] = [:] + if let usecols = usecols { + __kwargs["usecols"] = RemotePythonValue(usecols) + } + if let convert_categoricals = convert_categoricals { + __kwargs["convert_categoricals"] = RemotePythonValue(convert_categoricals) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_spss", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_spss", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func read_xml(path_or_buffer: String, xpath: String? = nil, namespaces: (any PythonConvertible)? = nil, elems_only: Bool? = nil, attrs_only: Bool? = nil, names: [String]? = nil, dtype: (any PythonConvertible)? = nil, encoding: String? = nil, parser: String? = nil, stylesheet: String? = nil, iterparse: (any PythonConvertible)? = nil, compression: String? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(path_or_buffer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let xpath = xpath { + __kwargs["xpath"] = RemotePythonValue(xpath) + } + if let namespaces = namespaces { + __kwargs["namespaces"] = RemotePythonValue(namespaces) + } + if let elems_only = elems_only { + __kwargs["elems_only"] = RemotePythonValue(elems_only) + } + if let attrs_only = attrs_only { + __kwargs["attrs_only"] = RemotePythonValue(attrs_only) + } + if let names = names { + __kwargs["names"] = RemotePythonValue(names) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let parser = parser { + __kwargs["parser"] = RemotePythonValue(parser) + } + if let stylesheet = stylesheet { + __kwargs["stylesheet"] = RemotePythonValue(stylesheet) + } + if let iterparse = iterparse { + __kwargs["iterparse"] = RemotePythonValue(iterparse) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_xml", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "read_xml", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func concat(objs: PyHandle, axis: Int? = nil, join: String? = nil, ignore_index: Bool? = nil, keys: PyHandle? = nil, levels: PyHandle? = nil, names: [String]? = nil, verify_integrity: Bool? = nil, sort: Bool? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(objs)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let join = join { + __kwargs["join"] = RemotePythonValue(join) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + if let keys = keys { + __kwargs["keys"] = RemotePythonValue(keys) + } + if let levels = levels { + __kwargs["levels"] = RemotePythonValue(levels) + } + if let names = names { + __kwargs["names"] = RemotePythonValue(names) + } + if let verify_integrity = verify_integrity { + __kwargs["verify_integrity"] = RemotePythonValue(verify_integrity) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "concat", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "concat", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func merge(left: any PythonConvertible, right: any PythonConvertible, how: String? = nil, on: (any PythonConvertible)? = nil, left_on: (any PythonConvertible)? = nil, right_on: (any PythonConvertible)? = nil, left_index: Bool? = nil, right_index: Bool? = nil, sort: Bool? = nil, suffixes: (any PythonConvertible)? = nil, copy: Bool? = nil, indicator: Bool? = nil, validate: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left)) + __args.append(RemotePythonValue(right)) + var __kwargs: [String: RemotePythonValue] = [:] + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + if let on = on { + __kwargs["on"] = RemotePythonValue(on) + } + if let left_on = left_on { + __kwargs["left_on"] = RemotePythonValue(left_on) + } + if let right_on = right_on { + __kwargs["right_on"] = RemotePythonValue(right_on) + } + if let left_index = left_index { + __kwargs["left_index"] = RemotePythonValue(left_index) + } + if let right_index = right_index { + __kwargs["right_index"] = RemotePythonValue(right_index) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let suffixes = suffixes { + __kwargs["suffixes"] = RemotePythonValue(suffixes) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let indicator = indicator { + __kwargs["indicator"] = RemotePythonValue(indicator) + } + if let validate = validate { + __kwargs["validate"] = RemotePythonValue(validate) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "merge", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "merge", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func merge_ordered(left: PoolDataFrame, right: PoolDataFrame, on: (any PythonConvertible)? = nil, left_on: (any PythonConvertible)? = nil, right_on: (any PythonConvertible)? = nil, left_by: (any PythonConvertible)? = nil, right_by: (any PythonConvertible)? = nil, fill_method: String? = nil, suffixes: (any PythonConvertible)? = nil, how: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left.handle)) + __args.append(RemotePythonValue(right.handle)) + if let on = on { + __args.append(RemotePythonValue(on)) + } + if let left_on = left_on { + __args.append(RemotePythonValue(left_on)) + } + if let right_on = right_on { + __args.append(RemotePythonValue(right_on)) + } + if let left_by = left_by { + __args.append(RemotePythonValue(left_by)) + } + if let right_by = right_by { + __args.append(RemotePythonValue(right_by)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let fill_method = fill_method { + __kwargs["fill_method"] = RemotePythonValue(fill_method) + } + if let suffixes = suffixes { + __kwargs["suffixes"] = RemotePythonValue(suffixes) + } + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "merge_ordered", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "merge_ordered", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func merge_asof(left: PoolDataFrame, right: PoolDataFrame, on: String? = nil, left_on: String? = nil, right_on: String? = nil, left_index: Bool? = nil, right_index: Bool? = nil, by: (any PythonConvertible)? = nil, left_by: String? = nil, right_by: String? = nil, suffixes: (any PythonConvertible)? = nil, tolerance: (any PythonConvertible)? = nil, allow_exact_matches: Bool? = nil, direction: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left.handle)) + __args.append(RemotePythonValue(right.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let on = on { + __kwargs["on"] = RemotePythonValue(on) + } + if let left_on = left_on { + __kwargs["left_on"] = RemotePythonValue(left_on) + } + if let right_on = right_on { + __kwargs["right_on"] = RemotePythonValue(right_on) + } + if let left_index = left_index { + __kwargs["left_index"] = RemotePythonValue(left_index) + } + if let right_index = right_index { + __kwargs["right_index"] = RemotePythonValue(right_index) + } + if let by = by { + __kwargs["by"] = RemotePythonValue(by) + } + if let left_by = left_by { + __kwargs["left_by"] = RemotePythonValue(left_by) + } + if let right_by = right_by { + __kwargs["right_by"] = RemotePythonValue(right_by) + } + if let suffixes = suffixes { + __kwargs["suffixes"] = RemotePythonValue(suffixes) + } + if let tolerance = tolerance { + __kwargs["tolerance"] = RemotePythonValue(tolerance) + } + if let allow_exact_matches = allow_exact_matches { + __kwargs["allow_exact_matches"] = RemotePythonValue(allow_exact_matches) + } + if let direction = direction { + __kwargs["direction"] = RemotePythonValue(direction) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "merge_asof", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "merge_asof", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pivot(data: PoolDataFrame, index: String? = nil, columns: String? = nil, values: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(data.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let values = values { + __kwargs["values"] = RemotePythonValue(values) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "pivot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "pivot", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pivot_table(data: PoolDataFrame, values: (any PythonConvertible)? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, aggfunc: String? = nil, fill_value: (any PythonConvertible)? = nil, margins: Bool? = nil, dropna: Bool? = nil, margins_name: String? = nil, observed: Bool? = nil, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(data.handle)) + if let values = values { + __args.append(RemotePythonValue(values)) + } + if let index = index { + __args.append(RemotePythonValue(index)) + } + if let columns = columns { + __args.append(RemotePythonValue(columns)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let aggfunc = aggfunc { + __kwargs["aggfunc"] = RemotePythonValue(aggfunc) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let margins = margins { + __kwargs["margins"] = RemotePythonValue(margins) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + if let margins_name = margins_name { + __kwargs["margins_name"] = RemotePythonValue(margins_name) + } + if let observed = observed { + __kwargs["observed"] = RemotePythonValue(observed) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "pivot_table", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "pivot_table", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func crosstab(index: any PythonConvertible, columns: any PythonConvertible, values: (any PythonConvertible)? = nil, rownames: [String]? = nil, colnames: [String]? = nil, aggfunc: String? = nil, margins: Bool? = nil, margins_name: String? = nil, dropna: Bool? = nil, normalize: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(index)) + __args.append(RemotePythonValue(columns)) + if let values = values { + __args.append(RemotePythonValue(values)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let rownames = rownames { + __kwargs["rownames"] = RemotePythonValue(rownames) + } + if let colnames = colnames { + __kwargs["colnames"] = RemotePythonValue(colnames) + } + if let aggfunc = aggfunc { + __kwargs["aggfunc"] = RemotePythonValue(aggfunc) + } + if let margins = margins { + __kwargs["margins"] = RemotePythonValue(margins) + } + if let margins_name = margins_name { + __kwargs["margins_name"] = RemotePythonValue(margins_name) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "crosstab", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "crosstab", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func melt(frame: PoolDataFrame, id_vars: (any PythonConvertible)? = nil, value_vars: (any PythonConvertible)? = nil, var_name: String? = nil, value_name: String? = nil, col_level: Int? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(frame.handle)) + if let id_vars = id_vars { + __args.append(RemotePythonValue(id_vars)) + } + if let value_vars = value_vars { + __args.append(RemotePythonValue(value_vars)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let var_name = var_name { + __kwargs["var_name"] = RemotePythonValue(var_name) + } + if let value_name = value_name { + __kwargs["value_name"] = RemotePythonValue(value_name) + } + if let col_level = col_level { + __kwargs["col_level"] = RemotePythonValue(col_level) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "melt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "melt", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func wide_to_long(df: PoolDataFrame, stubnames: any PythonConvertible, i: any PythonConvertible, j: String, sep: String? = nil, suffix: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(df.handle)) + __args.append(RemotePythonValue(stubnames)) + __args.append(RemotePythonValue(i)) + __args.append(RemotePythonValue(j)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let suffix = suffix { + __kwargs["suffix"] = RemotePythonValue(suffix) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "wide_to_long", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "wide_to_long", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func get_dummies(data: any PythonConvertible, prefix: (any PythonConvertible)? = nil, prefix_sep: String? = nil, dummy_na: Bool? = nil, columns: [String]? = nil, sparse: Bool? = nil, drop_first: Bool? = nil, dtype: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(data)) + if let prefix = prefix { + __args.append(RemotePythonValue(prefix)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let prefix_sep = prefix_sep { + __kwargs["prefix_sep"] = RemotePythonValue(prefix_sep) + } + if let dummy_na = dummy_na { + __kwargs["dummy_na"] = RemotePythonValue(dummy_na) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let sparse = sparse { + __kwargs["sparse"] = RemotePythonValue(sparse) + } + if let drop_first = drop_first { + __kwargs["drop_first"] = RemotePythonValue(drop_first) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "get_dummies", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "get_dummies", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cut(x: any PythonConvertible, bins: any PythonConvertible, right: Bool? = nil, labels: (any PythonConvertible)? = nil, retbins: Bool? = nil, precision: Int? = nil, include_lowest: Bool? = nil, duplicates: String? = nil, ordered: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(bins)) + var __kwargs: [String: RemotePythonValue] = [:] + if let right = right { + __kwargs["right"] = RemotePythonValue(right) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let retbins = retbins { + __kwargs["retbins"] = RemotePythonValue(retbins) + } + if let precision = precision { + __kwargs["precision"] = RemotePythonValue(precision) + } + if let include_lowest = include_lowest { + __kwargs["include_lowest"] = RemotePythonValue(include_lowest) + } + if let duplicates = duplicates { + __kwargs["duplicates"] = RemotePythonValue(duplicates) + } + if let ordered = ordered { + __kwargs["ordered"] = RemotePythonValue(ordered) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "cut", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "cut", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func qcut(x: any PythonConvertible, q: any PythonConvertible, labels: (any PythonConvertible)? = nil, retbins: Bool? = nil, precision: Int? = nil, duplicates: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(q)) + if let labels = labels { + __args.append(RemotePythonValue(labels)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let retbins = retbins { + __kwargs["retbins"] = RemotePythonValue(retbins) + } + if let precision = precision { + __kwargs["precision"] = RemotePythonValue(precision) + } + if let duplicates = duplicates { + __kwargs["duplicates"] = RemotePythonValue(duplicates) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "qcut", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "qcut", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func factorize(values: any PythonConvertible, sort: Bool? = nil, na_sentinel: Int? = nil, size_hint: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(values)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let na_sentinel = na_sentinel { + __kwargs["na_sentinel"] = RemotePythonValue(na_sentinel) + } + if let size_hint = size_hint { + __kwargs["size_hint"] = RemotePythonValue(size_hint) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "factorize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "factorize", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func unique(values: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(values)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "unique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "unique", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func isna(obj: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(obj)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "isna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "isna", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func isnull(obj: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(obj)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "isnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "isnull", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func notna(obj: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(obj)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "notna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "notna", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func notnull(obj: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(obj)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "notnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "notnull", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_numeric(arg: any PythonConvertible, errors: String? = nil, downcast: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arg)) + var __kwargs: [String: RemotePythonValue] = [:] + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + if let downcast = downcast { + __kwargs["downcast"] = RemotePythonValue(downcast) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "to_numeric", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "to_numeric", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_datetime(arg: any PythonConvertible, errors: String? = nil, dayfirst: Bool? = nil, yearfirst: Bool? = nil, utc: Bool? = nil, format: String? = nil, exact: Bool? = nil, unit: String? = nil, infer_datetime_format: Bool? = nil, origin: String? = nil, cache: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arg)) + var __kwargs: [String: RemotePythonValue] = [:] + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + if let dayfirst = dayfirst { + __kwargs["dayfirst"] = RemotePythonValue(dayfirst) + } + if let yearfirst = yearfirst { + __kwargs["yearfirst"] = RemotePythonValue(yearfirst) + } + if let utc = utc { + __kwargs["utc"] = RemotePythonValue(utc) + } + if let format = format { + __kwargs["format"] = RemotePythonValue(format) + } + if let exact = exact { + __kwargs["exact"] = RemotePythonValue(exact) + } + if let unit = unit { + __kwargs["unit"] = RemotePythonValue(unit) + } + if let infer_datetime_format = infer_datetime_format { + __kwargs["infer_datetime_format"] = RemotePythonValue(infer_datetime_format) + } + if let origin = origin { + __kwargs["origin"] = RemotePythonValue(origin) + } + if let cache = cache { + __kwargs["cache"] = RemotePythonValue(cache) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "to_datetime", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "to_datetime", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_timedelta(arg: any PythonConvertible, unit: String? = nil, errors: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arg)) + var __kwargs: [String: RemotePythonValue] = [:] + if let unit = unit { + __kwargs["unit"] = RemotePythonValue(unit) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "to_timedelta", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "to_timedelta", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func date_range(start: String? = nil, end: String? = nil, periods: Int? = nil, freq: String? = nil, tz: String? = nil, normalize: Bool? = nil, name: String? = nil, inclusive: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let start = start { + __kwargs["start"] = RemotePythonValue(start) + } + if let end = end { + __kwargs["end"] = RemotePythonValue(end) + } + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let freq = freq { + __kwargs["freq"] = RemotePythonValue(freq) + } + if let tz = tz { + __kwargs["tz"] = RemotePythonValue(tz) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let inclusive = inclusive { + __kwargs["inclusive"] = RemotePythonValue(inclusive) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "date_range", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "date_range", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func timedelta_range(start: String? = nil, end: String? = nil, periods: Int? = nil, freq: String? = nil, name: String? = nil, closed: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let start = start { + __kwargs["start"] = RemotePythonValue(start) + } + if let end = end { + __kwargs["end"] = RemotePythonValue(end) + } + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let freq = freq { + __kwargs["freq"] = RemotePythonValue(freq) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let closed = closed { + __kwargs["closed"] = RemotePythonValue(closed) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "timedelta_range", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "timedelta_range", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func period_range(start: String? = nil, end: String? = nil, periods: Int? = nil, freq: String? = nil, name: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let start = start { + __kwargs["start"] = RemotePythonValue(start) + } + if let end = end { + __kwargs["end"] = RemotePythonValue(end) + } + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let freq = freq { + __kwargs["freq"] = RemotePythonValue(freq) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "period_range", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "period_range", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func interval_range(start: (any PythonConvertible)? = nil, end: (any PythonConvertible)? = nil, periods: Int? = nil, freq: (any PythonConvertible)? = nil, name: String? = nil, closed: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + if let start = start { + __args.append(RemotePythonValue(start)) + } + if let end = end { + __args.append(RemotePythonValue(end)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let freq = freq { + __kwargs["freq"] = RemotePythonValue(freq) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let closed = closed { + __kwargs["closed"] = RemotePythonValue(closed) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "interval_range", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "interval_range", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func set_option(pat: String, value: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(pat)) + __args.append(RemotePythonValue(value)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + try await pool.invokeVoid(module: "pandas", function: "set_option", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "pandas", function: "set_option", args: __args, kwargs: __kwargs) + } + + public func get_option(pat: String, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(pat)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "get_option", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "get_option", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func reset_option(pat: String, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(pat)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + try await pool.invokeVoid(module: "pandas", function: "reset_option", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "pandas", function: "reset_option", args: __args, kwargs: __kwargs) + } + + public func describe_option(pat: String? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let pat = pat { + __kwargs["pat"] = RemotePythonValue(pat) + } + if let worker { + try await pool.invokeVoid(module: "pandas", function: "describe_option", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "pandas", function: "describe_option", args: __args, kwargs: __kwargs) + } + + public func option_context(_ args: [RemotePythonValue] = [], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "option_context", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "option_context", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func DataFrame(data: (any PythonConvertible)? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, dtype: String? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data = data { + __kwargs["data"] = RemotePythonValue(data) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "DataFrame", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "DataFrame", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func Series(data: (any PythonConvertible)? = nil, index: (any PythonConvertible)? = nil, dtype: String? = nil, name: String? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data = data { + __kwargs["data"] = RemotePythonValue(data) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "Series", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "Series", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func DataFrameGroupBy(worker: Int? = nil) async throws -> PoolDataFrameGroupBy { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "DataFrameGroupBy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "DataFrameGroupBy", args: __args, kwargs: __kwargs) + } + return PoolDataFrameGroupBy(pool: pool, owned: __owned) + } + + public func SeriesGroupBy(worker: Int? = nil) async throws -> PoolSeriesGroupBy { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "SeriesGroupBy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "SeriesGroupBy", args: __args, kwargs: __kwargs) + } + return PoolSeriesGroupBy(pool: pool, owned: __owned) + } + + public func Index(data: (any PythonConvertible)? = nil, dtype: String? = nil, copy: Bool? = nil, name: String? = nil, tupleize_cols: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data = data { + __kwargs["data"] = RemotePythonValue(data) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let tupleize_cols = tupleize_cols { + __kwargs["tupleize_cols"] = RemotePythonValue(tupleize_cols) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "pandas", function: "Index", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "pandas", function: "Index", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } +} + + +public struct PoolDataFrame: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func getItem(key: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(key)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "__getitem__", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "__getitem__", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func setItem(key: any PythonConvertible, value: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(key)) + __args.append(RemotePythonValue(value)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + _ = try await pool.methodOwned(handle: self, name: "__setitem__", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "__setitem__", args: __args, kwargs: __kwargs) + } + + public func head(n: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func tail(n: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func sample(n: Int? = nil, frac: Double? = nil, replace: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let frac = frac { + __kwargs["frac"] = RemotePythonValue(frac) + } + if let replace = replace { + __kwargs["replace"] = RemotePythonValue(replace) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sample", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sample", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func nlargest(n: Int, columns: any PythonConvertible, keep: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + __args.append(RemotePythonValue(columns)) + var __kwargs: [String: RemotePythonValue] = [:] + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nlargest", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nlargest", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func nsmallest(n: Int, columns: any PythonConvertible, keep: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + __args.append(RemotePythonValue(columns)) + var __kwargs: [String: RemotePythonValue] = [:] + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nsmallest", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nsmallest", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func query(expr: String, inplace: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(expr)) + var __kwargs: [String: RemotePythonValue] = [:] + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "query", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "query", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func filter(items: [String]? = nil, like: String? = nil, regex: String? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let items = items { + __kwargs["items"] = RemotePythonValue(items) + } + if let like = like { + __kwargs["like"] = RemotePythonValue(like) + } + if let regex = regex { + __kwargs["regex"] = RemotePythonValue(regex) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "filter", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "filter", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func info(verbose: Bool? = nil, buf: (any PythonConvertible)? = nil, max_cols: Int? = nil, memory_usage: Bool? = nil, show_counts: Bool? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let buf = buf { + __kwargs["buf"] = RemotePythonValue(buf) + } + if let max_cols = max_cols { + __kwargs["max_cols"] = RemotePythonValue(max_cols) + } + if let memory_usage = memory_usage { + __kwargs["memory_usage"] = RemotePythonValue(memory_usage) + } + if let show_counts = show_counts { + __kwargs["show_counts"] = RemotePythonValue(show_counts) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "info", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "info", args: __args, kwargs: __kwargs) + } + + public func describe(percentiles: [Double]? = nil, include: (any PythonConvertible)? = nil, exclude: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let percentiles = percentiles { + __kwargs["percentiles"] = RemotePythonValue(percentiles) + } + if let include = include { + __kwargs["include"] = RemotePythonValue(include) + } + if let exclude = exclude { + __kwargs["exclude"] = RemotePythonValue(exclude) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func memory_usage(index: Bool? = nil, deep: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "memory_usage", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "memory_usage", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func sum(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func mean(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func median(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func min(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func max(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func std(axis: Int? = nil, skipna: Bool? = nil, ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func `var`(axis: Int? = nil, skipna: Bool? = nil, ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func count(axis: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "count", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "count", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func nunique(axis: Int? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nunique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nunique", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func abs(worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "abs", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "abs", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func prod(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cumsum(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cumprod(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cummin(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cummax(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func diff(periods: Int? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pct_change(periods: Int? = nil, fill_method: String? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let fill_method = fill_method { + __kwargs["fill_method"] = RemotePythonValue(fill_method) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func corr(method: String? = nil, min_periods: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let min_periods = min_periods { + __kwargs["min_periods"] = RemotePythonValue(min_periods) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "corr", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "corr", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cov(min_periods: Int? = nil, ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let min_periods = min_periods { + __kwargs["min_periods"] = RemotePythonValue(min_periods) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cov", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cov", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func quantile(q: Double? = nil, axis: Int? = nil, numeric_only: Bool? = nil, interpolation: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let q = q { + __kwargs["q"] = RemotePythonValue(q) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let interpolation = interpolation { + __kwargs["interpolation"] = RemotePythonValue(interpolation) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func rank(axis: Int? = nil, method: String? = nil, numeric_only: Bool? = nil, na_option: String? = nil, ascending: Bool? = nil, pct: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let na_option = na_option { + __kwargs["na_option"] = RemotePythonValue(na_option) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let pct = pct { + __kwargs["pct"] = RemotePythonValue(pct) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func idxmax(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "idxmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "idxmax", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func idxmin(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "idxmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "idxmin", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func mode(axis: Int? = nil, numeric_only: Bool? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mode", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mode", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func value_counts(subset: [String]? = nil, normalize: Bool? = nil, sort: Bool? = nil, ascending: Bool? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "value_counts", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "value_counts", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func apply(`func`: any PythonConvertible, axis: Int? = nil, raw: Bool? = nil, result_type: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let raw = raw { + __kwargs["raw"] = RemotePythonValue(raw) + } + if let result_type = result_type { + __kwargs["result_type"] = RemotePythonValue(result_type) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func map(`func`: any PythonConvertible, na_action: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let na_action = na_action { + __kwargs["na_action"] = RemotePythonValue(na_action) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "map", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "map", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func transform(`func`: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func agg(`func`: String, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func aggregate(`func`: String, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func pipe(`func`: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pipe", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pipe", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func eval(expr: String, inplace: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(expr)) + var __kwargs: [String: RemotePythonValue] = [:] + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "eval", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "eval", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func assign(extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "assign", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "assign", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func clip(lower: Double? = nil, upper: Double? = nil, axis: Int? = nil, inplace: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let lower = lower { + __kwargs["lower"] = RemotePythonValue(lower) + } + if let upper = upper { + __kwargs["upper"] = RemotePythonValue(upper) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "clip", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "clip", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func round(decimals: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let decimals = decimals { + __kwargs["decimals"] = RemotePythonValue(decimals) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "round", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "round", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func groupby(by: (any PythonConvertible)? = nil, axis: Int? = nil, level: Int? = nil, as_index: Bool? = nil, sort: Bool? = nil, group_keys: Bool? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrameGroupBy { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let by = by { + __kwargs["by"] = RemotePythonValue(by) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let as_index = as_index { + __kwargs["as_index"] = RemotePythonValue(as_index) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let group_keys = group_keys { + __kwargs["group_keys"] = RemotePythonValue(group_keys) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "groupby", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "groupby", args: __args, kwargs: __kwargs) + } + return PoolDataFrameGroupBy(pool: pool, owned: __owned) + } + + public func sort_values(by: any PythonConvertible, axis: Int? = nil, ascending: Bool? = nil, inplace: Bool? = nil, kind: String? = nil, na_position: String? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(by)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let na_position = na_position { + __kwargs["na_position"] = RemotePythonValue(na_position) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sort_values", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sort_values", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func sort_index(axis: Int? = nil, level: Int? = nil, ascending: Bool? = nil, inplace: Bool? = nil, kind: String? = nil, na_position: String? = nil, sort_remaining: Bool? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let na_position = na_position { + __kwargs["na_position"] = RemotePythonValue(na_position) + } + if let sort_remaining = sort_remaining { + __kwargs["sort_remaining"] = RemotePythonValue(sort_remaining) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sort_index", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sort_index", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func isna(worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isna", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func isnull(worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isnull", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func notna(worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "notna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "notna", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func notnull(worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "notnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "notnull", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func dropna(axis: Int? = nil, how: String? = nil, thresh: Int? = nil, subset: [String]? = nil, inplace: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + if let thresh = thresh { + __kwargs["thresh"] = RemotePythonValue(thresh) + } + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "dropna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "dropna", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func fillna(value: (any PythonConvertible)? = nil, method: String? = nil, axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func ffill(axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func bfill(axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func interpolate(method: String? = nil, axis: Int? = nil, limit: Int? = nil, inplace: Bool? = nil, limit_direction: String? = nil, limit_area: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit_direction = limit_direction { + __kwargs["limit_direction"] = RemotePythonValue(limit_direction) + } + if let limit_area = limit_area { + __kwargs["limit_area"] = RemotePythonValue(limit_area) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "interpolate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "interpolate", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func replace(to_replace: (any PythonConvertible)? = nil, value: (any PythonConvertible)? = nil, inplace: Bool? = nil, limit: Int? = nil, regex: Bool? = nil, method: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let to_replace = to_replace { + __kwargs["to_replace"] = RemotePythonValue(to_replace) + } + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + if let regex = regex { + __kwargs["regex"] = RemotePythonValue(regex) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "replace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "replace", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func duplicated(subset: [String]? = nil, keep: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "duplicated", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "duplicated", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func drop_duplicates(subset: [String]? = nil, keep: String? = nil, inplace: Bool? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "drop_duplicates", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "drop_duplicates", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func transpose(copy: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transpose", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transpose", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pivot(index: String? = nil, columns: String? = nil, values: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let values = values { + __kwargs["values"] = RemotePythonValue(values) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pivot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pivot", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pivot_table(values: (any PythonConvertible)? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, aggfunc: String? = nil, fill_value: (any PythonConvertible)? = nil, margins: Bool? = nil, dropna: Bool? = nil, margins_name: String? = nil, observed: Bool? = nil, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let values = values { + __kwargs["values"] = RemotePythonValue(values) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let aggfunc = aggfunc { + __kwargs["aggfunc"] = RemotePythonValue(aggfunc) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let margins = margins { + __kwargs["margins"] = RemotePythonValue(margins) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + if let margins_name = margins_name { + __kwargs["margins_name"] = RemotePythonValue(margins_name) + } + if let observed = observed { + __kwargs["observed"] = RemotePythonValue(observed) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pivot_table", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pivot_table", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func melt(id_vars: (any PythonConvertible)? = nil, value_vars: (any PythonConvertible)? = nil, var_name: String? = nil, value_name: String? = nil, col_level: Int? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let id_vars = id_vars { + __kwargs["id_vars"] = RemotePythonValue(id_vars) + } + if let value_vars = value_vars { + __kwargs["value_vars"] = RemotePythonValue(value_vars) + } + if let var_name = var_name { + __kwargs["var_name"] = RemotePythonValue(var_name) + } + if let value_name = value_name { + __kwargs["value_name"] = RemotePythonValue(value_name) + } + if let col_level = col_level { + __kwargs["col_level"] = RemotePythonValue(col_level) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "melt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "melt", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func stack(level: Int? = nil, dropna: Bool? = nil, sort: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "stack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "stack", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func unstack(level: Int? = nil, fill_value: (any PythonConvertible)? = nil, sort: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "unstack", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "unstack", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func explode(column: String, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(column)) + var __kwargs: [String: RemotePythonValue] = [:] + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "explode", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "explode", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func merge(right: any PythonConvertible, how: String? = nil, on: (any PythonConvertible)? = nil, left_on: (any PythonConvertible)? = nil, right_on: (any PythonConvertible)? = nil, left_index: Bool? = nil, right_index: Bool? = nil, sort: Bool? = nil, suffixes: (any PythonConvertible)? = nil, copy: Bool? = nil, indicator: Bool? = nil, validate: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(right)) + var __kwargs: [String: RemotePythonValue] = [:] + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + if let on = on { + __kwargs["on"] = RemotePythonValue(on) + } + if let left_on = left_on { + __kwargs["left_on"] = RemotePythonValue(left_on) + } + if let right_on = right_on { + __kwargs["right_on"] = RemotePythonValue(right_on) + } + if let left_index = left_index { + __kwargs["left_index"] = RemotePythonValue(left_index) + } + if let right_index = right_index { + __kwargs["right_index"] = RemotePythonValue(right_index) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let suffixes = suffixes { + __kwargs["suffixes"] = RemotePythonValue(suffixes) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let indicator = indicator { + __kwargs["indicator"] = RemotePythonValue(indicator) + } + if let validate = validate { + __kwargs["validate"] = RemotePythonValue(validate) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "merge", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "merge", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func join(other: any PythonConvertible, on: (any PythonConvertible)? = nil, how: String? = nil, lsuffix: String? = nil, rsuffix: String? = nil, sort: Bool? = nil, validate: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let on = on { + __kwargs["on"] = RemotePythonValue(on) + } + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + if let lsuffix = lsuffix { + __kwargs["lsuffix"] = RemotePythonValue(lsuffix) + } + if let rsuffix = rsuffix { + __kwargs["rsuffix"] = RemotePythonValue(rsuffix) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let validate = validate { + __kwargs["validate"] = RemotePythonValue(validate) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "join", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "join", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func append(other: any PythonConvertible, ignore_index: Bool? = nil, verify_integrity: Bool? = nil, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + if let verify_integrity = verify_integrity { + __kwargs["verify_integrity"] = RemotePythonValue(verify_integrity) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "append", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "append", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func combine(other: PoolDataFrame, `func`: any PythonConvertible, fill_value: (any PythonConvertible)? = nil, overwrite: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let overwrite = overwrite { + __kwargs["overwrite"] = RemotePythonValue(overwrite) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "combine", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "combine", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func combine_first(other: PoolDataFrame, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "combine_first", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "combine_first", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func update(other: any PythonConvertible, join: String? = nil, overwrite: Bool? = nil, filter_func: (any PythonConvertible)? = nil, errors: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let join = join { + __kwargs["join"] = RemotePythonValue(join) + } + if let overwrite = overwrite { + __kwargs["overwrite"] = RemotePythonValue(overwrite) + } + if let filter_func = filter_func { + __kwargs["filter_func"] = RemotePythonValue(filter_func) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "update", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "update", args: __args, kwargs: __kwargs) + } + + public func drop(labels: (any PythonConvertible)? = nil, axis: Int? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, level: Int? = nil, inplace: Bool? = nil, errors: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "drop", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "drop", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func rename(mapper: (any PythonConvertible)? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, axis: Int? = nil, copy: Bool? = nil, inplace: Bool? = nil, level: Int? = nil, errors: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let mapper = mapper { + __kwargs["mapper"] = RemotePythonValue(mapper) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rename", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rename", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pop(item: String, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(item)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pop", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pop", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func insert(loc: Int, column: String, value: any PythonConvertible, allow_duplicates: Bool? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(loc)) + __args.append(RemotePythonValue(column)) + __args.append(RemotePythonValue(value)) + var __kwargs: [String: RemotePythonValue] = [:] + if let allow_duplicates = allow_duplicates { + __kwargs["allow_duplicates"] = RemotePythonValue(allow_duplicates) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "insert", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "insert", args: __args, kwargs: __kwargs) + } + + public func set_index(keys: any PythonConvertible, drop: Bool? = nil, append: Bool? = nil, inplace: Bool? = nil, verify_integrity: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(keys)) + var __kwargs: [String: RemotePythonValue] = [:] + if let drop = drop { + __kwargs["drop"] = RemotePythonValue(drop) + } + if let append = append { + __kwargs["append"] = RemotePythonValue(append) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let verify_integrity = verify_integrity { + __kwargs["verify_integrity"] = RemotePythonValue(verify_integrity) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "set_index", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "set_index", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func reset_index(level: Int? = nil, drop: Bool? = nil, inplace: Bool? = nil, col_level: Int? = nil, col_fill: String? = nil, allow_duplicates: Bool? = nil, names: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let drop = drop { + __kwargs["drop"] = RemotePythonValue(drop) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let col_level = col_level { + __kwargs["col_level"] = RemotePythonValue(col_level) + } + if let col_fill = col_fill { + __kwargs["col_fill"] = RemotePythonValue(col_fill) + } + if let allow_duplicates = allow_duplicates { + __kwargs["allow_duplicates"] = RemotePythonValue(allow_duplicates) + } + if let names = names { + __kwargs["names"] = RemotePythonValue(names) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "reset_index", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "reset_index", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func reindex(labels: (any PythonConvertible)? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, axis: Int? = nil, method: String? = nil, copy: Bool? = nil, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, limit: Int? = nil, tolerance: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + if let tolerance = tolerance { + __kwargs["tolerance"] = RemotePythonValue(tolerance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "reindex", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "reindex", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func astype(dtype: any PythonConvertible, copy: Bool? = nil, errors: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func convert_dtypes(infer_objects: Bool? = nil, convert_string: Bool? = nil, convert_integer: Bool? = nil, convert_boolean: Bool? = nil, convert_floating: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let infer_objects = infer_objects { + __kwargs["infer_objects"] = RemotePythonValue(infer_objects) + } + if let convert_string = convert_string { + __kwargs["convert_string"] = RemotePythonValue(convert_string) + } + if let convert_integer = convert_integer { + __kwargs["convert_integer"] = RemotePythonValue(convert_integer) + } + if let convert_boolean = convert_boolean { + __kwargs["convert_boolean"] = RemotePythonValue(convert_boolean) + } + if let convert_floating = convert_floating { + __kwargs["convert_floating"] = RemotePythonValue(convert_floating) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "convert_dtypes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "convert_dtypes", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func infer_objects(copy: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "infer_objects", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "infer_objects", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func copy(deep: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func any(axis: Int? = nil, bool_only: Bool? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let bool_only = bool_only { + __kwargs["bool_only"] = RemotePythonValue(bool_only) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "any", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "any", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func all(axis: Int? = nil, bool_only: Bool? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let bool_only = bool_only { + __kwargs["bool_only"] = RemotePythonValue(bool_only) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "all", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "all", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func eq(other: any PythonConvertible, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "eq", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "eq", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func ne(other: any PythonConvertible, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ne", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ne", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func lt(other: any PythonConvertible, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "lt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "lt", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func le(other: any PythonConvertible, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "le", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "le", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func gt(other: any PythonConvertible, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "gt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "gt", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func ge(other: any PythonConvertible, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ge", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ge", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func isin(values: any PythonConvertible, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(values)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isin", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func `where`(cond: any PythonConvertible, other: (any PythonConvertible)? = nil, inplace: Bool? = nil, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(cond)) + var __kwargs: [String: RemotePythonValue] = [:] + if let other = other { + __kwargs["other"] = RemotePythonValue(other) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "where", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "where", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func mask(cond: any PythonConvertible, other: (any PythonConvertible)? = nil, inplace: Bool? = nil, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(cond)) + var __kwargs: [String: RemotePythonValue] = [:] + if let other = other { + __kwargs["other"] = RemotePythonValue(other) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mask", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mask", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func to_csv(path_or_buf: String? = nil, sep: String? = nil, na_rep: String? = nil, float_format: String? = nil, columns: [String]? = nil, header: Bool? = nil, index: Bool? = nil, index_label: String? = nil, mode: String? = nil, encoding: String? = nil, compression: String? = nil, quoting: Int? = nil, quotechar: String? = nil, lineterminator: String? = nil, chunksize: Int? = nil, date_format: String? = nil, doublequote: Bool? = nil, escapechar: String? = nil, decimal: String? = nil, errors: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let path_or_buf = path_or_buf { + __kwargs["path_or_buf"] = RemotePythonValue(path_or_buf) + } + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let index_label = index_label { + __kwargs["index_label"] = RemotePythonValue(index_label) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let quoting = quoting { + __kwargs["quoting"] = RemotePythonValue(quoting) + } + if let quotechar = quotechar { + __kwargs["quotechar"] = RemotePythonValue(quotechar) + } + if let lineterminator = lineterminator { + __kwargs["lineterminator"] = RemotePythonValue(lineterminator) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let date_format = date_format { + __kwargs["date_format"] = RemotePythonValue(date_format) + } + if let doublequote = doublequote { + __kwargs["doublequote"] = RemotePythonValue(doublequote) + } + if let escapechar = escapechar { + __kwargs["escapechar"] = RemotePythonValue(escapechar) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_csv", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_csv", args: __args, kwargs: __kwargs) + } + + public func to_excel(excel_writer: any PythonConvertible, sheet_name: String? = nil, na_rep: String? = nil, float_format: String? = nil, columns: [String]? = nil, header: Bool? = nil, index: Bool? = nil, index_label: String? = nil, startrow: Int? = nil, startcol: Int? = nil, engine: String? = nil, merge_cells: Bool? = nil, inf_rep: String? = nil, freeze_panes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(excel_writer)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sheet_name = sheet_name { + __kwargs["sheet_name"] = RemotePythonValue(sheet_name) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let index_label = index_label { + __kwargs["index_label"] = RemotePythonValue(index_label) + } + if let startrow = startrow { + __kwargs["startrow"] = RemotePythonValue(startrow) + } + if let startcol = startcol { + __kwargs["startcol"] = RemotePythonValue(startcol) + } + if let engine = engine { + __kwargs["engine"] = RemotePythonValue(engine) + } + if let merge_cells = merge_cells { + __kwargs["merge_cells"] = RemotePythonValue(merge_cells) + } + if let inf_rep = inf_rep { + __kwargs["inf_rep"] = RemotePythonValue(inf_rep) + } + if let freeze_panes = freeze_panes { + __kwargs["freeze_panes"] = RemotePythonValue(freeze_panes) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "to_excel", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "to_excel", args: __args, kwargs: __kwargs) + } + + public func to_json(path_or_buf: String? = nil, orient: String? = nil, date_format: String? = nil, double_precision: Int? = nil, force_ascii: Bool? = nil, date_unit: String? = nil, default_handler: (any PythonConvertible)? = nil, lines: Bool? = nil, compression: String? = nil, index: Bool? = nil, indent: Int? = nil, mode: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let path_or_buf = path_or_buf { + __kwargs["path_or_buf"] = RemotePythonValue(path_or_buf) + } + if let orient = orient { + __kwargs["orient"] = RemotePythonValue(orient) + } + if let date_format = date_format { + __kwargs["date_format"] = RemotePythonValue(date_format) + } + if let double_precision = double_precision { + __kwargs["double_precision"] = RemotePythonValue(double_precision) + } + if let force_ascii = force_ascii { + __kwargs["force_ascii"] = RemotePythonValue(force_ascii) + } + if let date_unit = date_unit { + __kwargs["date_unit"] = RemotePythonValue(date_unit) + } + if let default_handler = default_handler { + __kwargs["default_handler"] = RemotePythonValue(default_handler) + } + if let lines = lines { + __kwargs["lines"] = RemotePythonValue(lines) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let indent = indent { + __kwargs["indent"] = RemotePythonValue(indent) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_json", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_json", args: __args, kwargs: __kwargs) + } + + public func to_html(buf: (any PythonConvertible)? = nil, columns: [String]? = nil, col_space: (any PythonConvertible)? = nil, header: Bool? = nil, index: Bool? = nil, na_rep: String? = nil, formatters: (any PythonConvertible)? = nil, float_format: String? = nil, sparsify: Bool? = nil, index_names: Bool? = nil, justify: String? = nil, max_rows: Int? = nil, max_cols: Int? = nil, show_dimensions: Bool? = nil, decimal: String? = nil, bold_rows: Bool? = nil, classes: (any PythonConvertible)? = nil, escape: Bool? = nil, notebook: Bool? = nil, border: Int? = nil, table_id: String? = nil, render_links: Bool? = nil, encoding: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let buf = buf { + __kwargs["buf"] = RemotePythonValue(buf) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let col_space = col_space { + __kwargs["col_space"] = RemotePythonValue(col_space) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let formatters = formatters { + __kwargs["formatters"] = RemotePythonValue(formatters) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let sparsify = sparsify { + __kwargs["sparsify"] = RemotePythonValue(sparsify) + } + if let index_names = index_names { + __kwargs["index_names"] = RemotePythonValue(index_names) + } + if let justify = justify { + __kwargs["justify"] = RemotePythonValue(justify) + } + if let max_rows = max_rows { + __kwargs["max_rows"] = RemotePythonValue(max_rows) + } + if let max_cols = max_cols { + __kwargs["max_cols"] = RemotePythonValue(max_cols) + } + if let show_dimensions = show_dimensions { + __kwargs["show_dimensions"] = RemotePythonValue(show_dimensions) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let bold_rows = bold_rows { + __kwargs["bold_rows"] = RemotePythonValue(bold_rows) + } + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + if let escape = escape { + __kwargs["escape"] = RemotePythonValue(escape) + } + if let notebook = notebook { + __kwargs["notebook"] = RemotePythonValue(notebook) + } + if let border = border { + __kwargs["border"] = RemotePythonValue(border) + } + if let table_id = table_id { + __kwargs["table_id"] = RemotePythonValue(table_id) + } + if let render_links = render_links { + __kwargs["render_links"] = RemotePythonValue(render_links) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_html", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_html", args: __args, kwargs: __kwargs) + } + + public func to_dict(orient: String? = nil, into: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let orient = orient { + __kwargs["orient"] = RemotePythonValue(orient) + } + if let into = into { + __kwargs["into"] = RemotePythonValue(into) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_dict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_dict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_numpy(dtype: String? = nil, copy: Bool? = nil, na_value: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let na_value = na_value { + __kwargs["na_value"] = RemotePythonValue(na_value) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_numpy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_numpy", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_records(index: Bool? = nil, column_dtypes: (any PythonConvertible)? = nil, index_dtypes: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let column_dtypes = column_dtypes { + __kwargs["column_dtypes"] = RemotePythonValue(column_dtypes) + } + if let index_dtypes = index_dtypes { + __kwargs["index_dtypes"] = RemotePythonValue(index_dtypes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_records", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_records", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_string(buf: (any PythonConvertible)? = nil, columns: [String]? = nil, col_space: Int? = nil, header: Bool? = nil, index: Bool? = nil, na_rep: String? = nil, formatters: (any PythonConvertible)? = nil, float_format: String? = nil, sparsify: Bool? = nil, index_names: Bool? = nil, justify: String? = nil, max_rows: Int? = nil, max_cols: Int? = nil, show_dimensions: Bool? = nil, decimal: String? = nil, line_width: Int? = nil, min_rows: Int? = nil, max_colwidth: Int? = nil, encoding: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let buf = buf { + __kwargs["buf"] = RemotePythonValue(buf) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let col_space = col_space { + __kwargs["col_space"] = RemotePythonValue(col_space) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let formatters = formatters { + __kwargs["formatters"] = RemotePythonValue(formatters) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let sparsify = sparsify { + __kwargs["sparsify"] = RemotePythonValue(sparsify) + } + if let index_names = index_names { + __kwargs["index_names"] = RemotePythonValue(index_names) + } + if let justify = justify { + __kwargs["justify"] = RemotePythonValue(justify) + } + if let max_rows = max_rows { + __kwargs["max_rows"] = RemotePythonValue(max_rows) + } + if let max_cols = max_cols { + __kwargs["max_cols"] = RemotePythonValue(max_cols) + } + if let show_dimensions = show_dimensions { + __kwargs["show_dimensions"] = RemotePythonValue(show_dimensions) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let line_width = line_width { + __kwargs["line_width"] = RemotePythonValue(line_width) + } + if let min_rows = min_rows { + __kwargs["min_rows"] = RemotePythonValue(min_rows) + } + if let max_colwidth = max_colwidth { + __kwargs["max_colwidth"] = RemotePythonValue(max_colwidth) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_string", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_string", args: __args, kwargs: __kwargs) + } + + public func to_markdown(buf: (any PythonConvertible)? = nil, mode: String? = nil, index: Bool? = nil, storage_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let buf = buf { + __kwargs["buf"] = RemotePythonValue(buf) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let storage_options = storage_options { + __kwargs["storage_options"] = RemotePythonValue(storage_options) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_markdown", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_markdown", args: __args, kwargs: __kwargs) + } + + public func to_parquet(path: String? = nil, engine: String? = nil, compression: String? = nil, index: Bool? = nil, partition_cols: [String]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let path = path { + __kwargs["path"] = RemotePythonValue(path) + } + if let engine = engine { + __kwargs["engine"] = RemotePythonValue(engine) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let partition_cols = partition_cols { + __kwargs["partition_cols"] = RemotePythonValue(partition_cols) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_parquet", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_parquet", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_sql(name: String, con: any PythonConvertible, schema: String? = nil, if_exists: String? = nil, index: Bool? = nil, index_label: (any PythonConvertible)? = nil, chunksize: Int? = nil, dtype: (any PythonConvertible)? = nil, method: String? = nil, worker: Int? = nil) async throws -> Int? { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(name)) + __args.append(RemotePythonValue(con)) + var __kwargs: [String: RemotePythonValue] = [:] + if let schema = schema { + __kwargs["schema"] = RemotePythonValue(schema) + } + if let if_exists = if_exists { + __kwargs["if_exists"] = RemotePythonValue(if_exists) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let index_label = index_label { + __kwargs["index_label"] = RemotePythonValue(index_label) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_sql", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_sql", args: __args, kwargs: __kwargs) + } + + public func to_pickle(path: String, compression: String? = nil, `protocol`: Int? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(path)) + var __kwargs: [String: RemotePythonValue] = [:] + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let `protocol` = `protocol` { + __kwargs["protocol"] = RemotePythonValue(`protocol`) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "to_pickle", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "to_pickle", args: __args, kwargs: __kwargs) + } + + public func to_clipboard(excel: Bool? = nil, sep: String? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let excel = excel { + __kwargs["excel"] = RemotePythonValue(excel) + } + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let worker { + _ = try await pool.methodOwned(handle: self, name: "to_clipboard", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "to_clipboard", args: __args, kwargs: __kwargs) + } + + public func to_latex(buf: (any PythonConvertible)? = nil, columns: [String]? = nil, col_space: Int? = nil, header: Bool? = nil, index: Bool? = nil, na_rep: String? = nil, formatters: (any PythonConvertible)? = nil, float_format: String? = nil, sparsify: Bool? = nil, index_names: Bool? = nil, bold_rows: Bool? = nil, column_format: String? = nil, longtable: Bool? = nil, escape: Bool? = nil, encoding: String? = nil, decimal: String? = nil, multicolumn: Bool? = nil, multicolumn_format: String? = nil, multirow: Bool? = nil, caption: String? = nil, label: String? = nil, position: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let buf = buf { + __kwargs["buf"] = RemotePythonValue(buf) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let col_space = col_space { + __kwargs["col_space"] = RemotePythonValue(col_space) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let formatters = formatters { + __kwargs["formatters"] = RemotePythonValue(formatters) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let sparsify = sparsify { + __kwargs["sparsify"] = RemotePythonValue(sparsify) + } + if let index_names = index_names { + __kwargs["index_names"] = RemotePythonValue(index_names) + } + if let bold_rows = bold_rows { + __kwargs["bold_rows"] = RemotePythonValue(bold_rows) + } + if let column_format = column_format { + __kwargs["column_format"] = RemotePythonValue(column_format) + } + if let longtable = longtable { + __kwargs["longtable"] = RemotePythonValue(longtable) + } + if let escape = escape { + __kwargs["escape"] = RemotePythonValue(escape) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let multicolumn = multicolumn { + __kwargs["multicolumn"] = RemotePythonValue(multicolumn) + } + if let multicolumn_format = multicolumn_format { + __kwargs["multicolumn_format"] = RemotePythonValue(multicolumn_format) + } + if let multirow = multirow { + __kwargs["multirow"] = RemotePythonValue(multirow) + } + if let caption = caption { + __kwargs["caption"] = RemotePythonValue(caption) + } + if let label = label { + __kwargs["label"] = RemotePythonValue(label) + } + if let position = position { + __kwargs["position"] = RemotePythonValue(position) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_latex", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_latex", args: __args, kwargs: __kwargs) + } + + public func items(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "items", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "items", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func iterrows(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "iterrows", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "iterrows", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func itertuples(index: Bool? = nil, name: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "itertuples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "itertuples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func keys(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "keys", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "keys", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func equals(other: PoolDataFrame, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "equals", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "equals", args: __args, kwargs: __kwargs) + } + + public func compare(other: PoolDataFrame, align_axis: Int? = nil, keep_shape: Bool? = nil, keep_equal: Bool? = nil, result_names: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let align_axis = align_axis { + __kwargs["align_axis"] = RemotePythonValue(align_axis) + } + if let keep_shape = keep_shape { + __kwargs["keep_shape"] = RemotePythonValue(keep_shape) + } + if let keep_equal = keep_equal { + __kwargs["keep_equal"] = RemotePythonValue(keep_equal) + } + if let result_names = result_names { + __kwargs["result_names"] = RemotePythonValue(result_names) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "compare", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "compare", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } +} + + +public struct PoolSeries: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func getItem(key: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(key)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "__getitem__", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "__getitem__", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func setItem(key: any PythonConvertible, value: any PythonConvertible, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(key)) + __args.append(RemotePythonValue(value)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + _ = try await pool.methodOwned(handle: self, name: "__setitem__", args: __args, kwargs: __kwargs, worker: worker) + return + } + _ = try await pool.methodOwned(handle: self, name: "__setitem__", args: __args, kwargs: __kwargs) + } + + public func head(n: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func tail(n: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func sample(n: Int? = nil, frac: Double? = nil, replace: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let frac = frac { + __kwargs["frac"] = RemotePythonValue(frac) + } + if let replace = replace { + __kwargs["replace"] = RemotePythonValue(replace) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sample", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sample", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func nlargest(n: Int? = nil, keep: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nlargest", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nlargest", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func nsmallest(n: Int? = nil, keep: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nsmallest", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nsmallest", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func describe(percentiles: [Double]? = nil, include: (any PythonConvertible)? = nil, exclude: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let percentiles = percentiles { + __kwargs["percentiles"] = RemotePythonValue(percentiles) + } + if let include = include { + __kwargs["include"] = RemotePythonValue(include) + } + if let exclude = exclude { + __kwargs["exclude"] = RemotePythonValue(exclude) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func memory_usage(index: Bool? = nil, deep: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + if let worker { + return try await pool.methodResult(handle: self, name: "memory_usage", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "memory_usage", args: __args, kwargs: __kwargs) + } + + public func sum(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func mean(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func median(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func min(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func max(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func std(axis: Int? = nil, skipna: Bool? = nil, ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func `var`(axis: Int? = nil, skipna: Bool? = nil, ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func count(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "count", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "count", args: __args, kwargs: __kwargs) + } + + public func nunique(dropna: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "nunique", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "nunique", args: __args, kwargs: __kwargs) + } + + public func value_counts(normalize: Bool? = nil, sort: Bool? = nil, ascending: Bool? = nil, bins: Int? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let bins = bins { + __kwargs["bins"] = RemotePythonValue(bins) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "value_counts", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "value_counts", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func unique(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "unique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "unique", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func abs(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "abs", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "abs", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func prod(axis: Int? = nil, skipna: Bool? = nil, numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cumsum(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cumprod(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cummin(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cummax(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func diff(periods: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func pct_change(periods: Int? = nil, fill_method: String? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let fill_method = fill_method { + __kwargs["fill_method"] = RemotePythonValue(fill_method) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func corr(other: PoolSeries, method: String? = nil, min_periods: Int? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let min_periods = min_periods { + __kwargs["min_periods"] = RemotePythonValue(min_periods) + } + if let worker { + return try await pool.methodResult(handle: self, name: "corr", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "corr", args: __args, kwargs: __kwargs) + } + + public func cov(other: PoolSeries, min_periods: Int? = nil, ddof: Int? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let min_periods = min_periods { + __kwargs["min_periods"] = RemotePythonValue(min_periods) + } + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let worker { + return try await pool.methodResult(handle: self, name: "cov", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "cov", args: __args, kwargs: __kwargs) + } + + public func quantile(q: Double? = nil, interpolation: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let q = q { + __kwargs["q"] = RemotePythonValue(q) + } + if let interpolation = interpolation { + __kwargs["interpolation"] = RemotePythonValue(interpolation) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func rank(axis: Int? = nil, method: String? = nil, numeric_only: Bool? = nil, na_option: String? = nil, ascending: Bool? = nil, pct: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let na_option = na_option { + __kwargs["na_option"] = RemotePythonValue(na_option) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let pct = pct { + __kwargs["pct"] = RemotePythonValue(pct) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func idxmax(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "idxmax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "idxmax", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func idxmin(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "idxmin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "idxmin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func mode(dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mode", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mode", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func argmax(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "argmax", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "argmax", args: __args, kwargs: __kwargs) + } + + public func argmin(axis: Int? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "argmin", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "argmin", args: __args, kwargs: __kwargs) + } + + public func apply(`func`: any PythonConvertible, convert_dtype: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let convert_dtype = convert_dtype { + __kwargs["convert_dtype"] = RemotePythonValue(convert_dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func map(arg: any PythonConvertible, na_action: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(arg)) + var __kwargs: [String: RemotePythonValue] = [:] + if let na_action = na_action { + __kwargs["na_action"] = RemotePythonValue(na_action) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "map", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "map", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func transform(`func`: any PythonConvertible, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func agg(`func`: String, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func aggregate(`func`: String, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func pipe(`func`: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pipe", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pipe", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func clip(lower: Double? = nil, upper: Double? = nil, axis: Int? = nil, inplace: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let lower = lower { + __kwargs["lower"] = RemotePythonValue(lower) + } + if let upper = upper { + __kwargs["upper"] = RemotePythonValue(upper) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "clip", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "clip", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func round(decimals: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let decimals = decimals { + __kwargs["decimals"] = RemotePythonValue(decimals) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "round", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "round", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func groupby(by: (any PythonConvertible)? = nil, axis: Int? = nil, level: Int? = nil, as_index: Bool? = nil, sort: Bool? = nil, group_keys: Bool? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeriesGroupBy { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let by = by { + __kwargs["by"] = RemotePythonValue(by) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let as_index = as_index { + __kwargs["as_index"] = RemotePythonValue(as_index) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let group_keys = group_keys { + __kwargs["group_keys"] = RemotePythonValue(group_keys) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "groupby", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "groupby", args: __args, kwargs: __kwargs) + } + return PoolSeriesGroupBy(pool: pool, owned: __owned) + } + + public func sort_values(axis: Int? = nil, ascending: Bool? = nil, inplace: Bool? = nil, kind: String? = nil, na_position: String? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let na_position = na_position { + __kwargs["na_position"] = RemotePythonValue(na_position) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sort_values", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sort_values", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func sort_index(axis: Int? = nil, level: Int? = nil, ascending: Bool? = nil, inplace: Bool? = nil, kind: String? = nil, na_position: String? = nil, sort_remaining: Bool? = nil, ignore_index: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + if let na_position = na_position { + __kwargs["na_position"] = RemotePythonValue(na_position) + } + if let sort_remaining = sort_remaining { + __kwargs["sort_remaining"] = RemotePythonValue(sort_remaining) + } + if let ignore_index = ignore_index { + __kwargs["ignore_index"] = RemotePythonValue(ignore_index) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sort_index", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sort_index", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func argsort(axis: Int? = nil, kind: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let kind = kind { + __kwargs["kind"] = RemotePythonValue(kind) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "argsort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "argsort", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func isna(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isna", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func isnull(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isnull", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func notna(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "notna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "notna", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func notnull(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "notnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "notnull", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func dropna(axis: Int? = nil, inplace: Bool? = nil, how: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "dropna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "dropna", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func fillna(value: (any PythonConvertible)? = nil, method: String? = nil, axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func ffill(axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func bfill(axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func interpolate(method: String? = nil, axis: Int? = nil, limit: Int? = nil, inplace: Bool? = nil, limit_direction: String? = nil, limit_area: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit_direction = limit_direction { + __kwargs["limit_direction"] = RemotePythonValue(limit_direction) + } + if let limit_area = limit_area { + __kwargs["limit_area"] = RemotePythonValue(limit_area) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "interpolate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "interpolate", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func replace(to_replace: (any PythonConvertible)? = nil, value: (any PythonConvertible)? = nil, inplace: Bool? = nil, limit: Int? = nil, regex: Bool? = nil, method: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let to_replace = to_replace { + __kwargs["to_replace"] = RemotePythonValue(to_replace) + } + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + if let regex = regex { + __kwargs["regex"] = RemotePythonValue(regex) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "replace", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "replace", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func duplicated(keep: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "duplicated", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "duplicated", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func drop_duplicates(keep: String? = nil, inplace: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "drop_duplicates", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "drop_duplicates", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func reset_index(level: Int? = nil, drop: Bool? = nil, name: String? = nil, inplace: Bool? = nil, allow_duplicates: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let drop = drop { + __kwargs["drop"] = RemotePythonValue(drop) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let allow_duplicates = allow_duplicates { + __kwargs["allow_duplicates"] = RemotePythonValue(allow_duplicates) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "reset_index", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "reset_index", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func reindex(index: (any PythonConvertible)? = nil, axis: Int? = nil, method: String? = nil, copy: Bool? = nil, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, limit: Int? = nil, tolerance: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + if let tolerance = tolerance { + __kwargs["tolerance"] = RemotePythonValue(tolerance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "reindex", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "reindex", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func rename(index: (any PythonConvertible)? = nil, axis: Int? = nil, copy: Bool? = nil, inplace: Bool? = nil, level: Int? = nil, errors: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rename", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rename", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func drop(labels: (any PythonConvertible)? = nil, axis: Int? = nil, index: (any PythonConvertible)? = nil, columns: (any PythonConvertible)? = nil, level: Int? = nil, inplace: Bool? = nil, errors: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let columns = columns { + __kwargs["columns"] = RemotePythonValue(columns) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "drop", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "drop", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func astype(dtype: any PythonConvertible, copy: Bool? = nil, errors: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func convert_dtypes(infer_objects: Bool? = nil, convert_string: Bool? = nil, convert_integer: Bool? = nil, convert_boolean: Bool? = nil, convert_floating: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let infer_objects = infer_objects { + __kwargs["infer_objects"] = RemotePythonValue(infer_objects) + } + if let convert_string = convert_string { + __kwargs["convert_string"] = RemotePythonValue(convert_string) + } + if let convert_integer = convert_integer { + __kwargs["convert_integer"] = RemotePythonValue(convert_integer) + } + if let convert_boolean = convert_boolean { + __kwargs["convert_boolean"] = RemotePythonValue(convert_boolean) + } + if let convert_floating = convert_floating { + __kwargs["convert_floating"] = RemotePythonValue(convert_floating) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "convert_dtypes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "convert_dtypes", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func infer_objects(copy: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "infer_objects", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "infer_objects", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func copy(deep: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func any(axis: Int? = nil, bool_only: Bool? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> Bool { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let bool_only = bool_only { + __kwargs["bool_only"] = RemotePythonValue(bool_only) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "any", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "any", args: __args, kwargs: __kwargs) + } + + public func all(axis: Int? = nil, bool_only: Bool? = nil, skipna: Bool? = nil, worker: Int? = nil) async throws -> Bool { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let bool_only = bool_only { + __kwargs["bool_only"] = RemotePythonValue(bool_only) + } + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "all", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "all", args: __args, kwargs: __kwargs) + } + + public func eq(other: any PythonConvertible, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "eq", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "eq", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func ne(other: any PythonConvertible, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ne", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ne", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func lt(other: any PythonConvertible, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "lt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "lt", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func le(other: any PythonConvertible, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "le", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "le", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func gt(other: any PythonConvertible, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "gt", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "gt", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func ge(other: any PythonConvertible, level: Int? = nil, fill_value: (any PythonConvertible)? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ge", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ge", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func isin(values: any PythonConvertible, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(values)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isin", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func between(left: any PythonConvertible, right: any PythonConvertible, inclusive: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left)) + __args.append(RemotePythonValue(right)) + var __kwargs: [String: RemotePythonValue] = [:] + if let inclusive = inclusive { + __kwargs["inclusive"] = RemotePythonValue(inclusive) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "between", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "between", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func `where`(cond: any PythonConvertible, other: (any PythonConvertible)? = nil, inplace: Bool? = nil, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(cond)) + var __kwargs: [String: RemotePythonValue] = [:] + if let other = other { + __kwargs["other"] = RemotePythonValue(other) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "where", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "where", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func mask(cond: any PythonConvertible, other: (any PythonConvertible)? = nil, inplace: Bool? = nil, axis: Int? = nil, level: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(cond)) + var __kwargs: [String: RemotePythonValue] = [:] + if let other = other { + __kwargs["other"] = RemotePythonValue(other) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mask", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mask", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func to_csv(path_or_buf: String? = nil, sep: String? = nil, na_rep: String? = nil, float_format: String? = nil, header: Bool? = nil, index: Bool? = nil, index_label: String? = nil, mode: String? = nil, encoding: String? = nil, compression: String? = nil, quoting: Int? = nil, quotechar: String? = nil, lineterminator: String? = nil, chunksize: Int? = nil, date_format: String? = nil, doublequote: Bool? = nil, escapechar: String? = nil, decimal: String? = nil, errors: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let path_or_buf = path_or_buf { + __kwargs["path_or_buf"] = RemotePythonValue(path_or_buf) + } + if let sep = sep { + __kwargs["sep"] = RemotePythonValue(sep) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let index_label = index_label { + __kwargs["index_label"] = RemotePythonValue(index_label) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let encoding = encoding { + __kwargs["encoding"] = RemotePythonValue(encoding) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let quoting = quoting { + __kwargs["quoting"] = RemotePythonValue(quoting) + } + if let quotechar = quotechar { + __kwargs["quotechar"] = RemotePythonValue(quotechar) + } + if let lineterminator = lineterminator { + __kwargs["lineterminator"] = RemotePythonValue(lineterminator) + } + if let chunksize = chunksize { + __kwargs["chunksize"] = RemotePythonValue(chunksize) + } + if let date_format = date_format { + __kwargs["date_format"] = RemotePythonValue(date_format) + } + if let doublequote = doublequote { + __kwargs["doublequote"] = RemotePythonValue(doublequote) + } + if let escapechar = escapechar { + __kwargs["escapechar"] = RemotePythonValue(escapechar) + } + if let decimal = decimal { + __kwargs["decimal"] = RemotePythonValue(decimal) + } + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_csv", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_csv", args: __args, kwargs: __kwargs) + } + + public func to_dict(into: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let into = into { + __kwargs["into"] = RemotePythonValue(into) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_dict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_dict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_list(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_list", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_list", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func tolist(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tolist", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tolist", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_numpy(dtype: String? = nil, copy: Bool? = nil, na_value: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let na_value = na_value { + __kwargs["na_value"] = RemotePythonValue(na_value) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_numpy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_numpy", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_frame(name: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_frame", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_frame", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func to_string(buf: (any PythonConvertible)? = nil, na_rep: String? = nil, float_format: String? = nil, header: Bool? = nil, index: Bool? = nil, length: Bool? = nil, dtype: Bool? = nil, name: Bool? = nil, max_rows: Int? = nil, min_rows: Int? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let buf = buf { + __kwargs["buf"] = RemotePythonValue(buf) + } + if let na_rep = na_rep { + __kwargs["na_rep"] = RemotePythonValue(na_rep) + } + if let float_format = float_format { + __kwargs["float_format"] = RemotePythonValue(float_format) + } + if let header = header { + __kwargs["header"] = RemotePythonValue(header) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let length = length { + __kwargs["length"] = RemotePythonValue(length) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let max_rows = max_rows { + __kwargs["max_rows"] = RemotePythonValue(max_rows) + } + if let min_rows = min_rows { + __kwargs["min_rows"] = RemotePythonValue(min_rows) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_string", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_string", args: __args, kwargs: __kwargs) + } + + public func to_json(path_or_buf: String? = nil, orient: String? = nil, date_format: String? = nil, double_precision: Int? = nil, force_ascii: Bool? = nil, date_unit: String? = nil, default_handler: (any PythonConvertible)? = nil, lines: Bool? = nil, compression: String? = nil, index: Bool? = nil, indent: Int? = nil, mode: String? = nil, worker: Int? = nil) async throws -> String? { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let path_or_buf = path_or_buf { + __kwargs["path_or_buf"] = RemotePythonValue(path_or_buf) + } + if let orient = orient { + __kwargs["orient"] = RemotePythonValue(orient) + } + if let date_format = date_format { + __kwargs["date_format"] = RemotePythonValue(date_format) + } + if let double_precision = double_precision { + __kwargs["double_precision"] = RemotePythonValue(double_precision) + } + if let force_ascii = force_ascii { + __kwargs["force_ascii"] = RemotePythonValue(force_ascii) + } + if let date_unit = date_unit { + __kwargs["date_unit"] = RemotePythonValue(date_unit) + } + if let default_handler = default_handler { + __kwargs["default_handler"] = RemotePythonValue(default_handler) + } + if let lines = lines { + __kwargs["lines"] = RemotePythonValue(lines) + } + if let compression = compression { + __kwargs["compression"] = RemotePythonValue(compression) + } + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let indent = indent { + __kwargs["indent"] = RemotePythonValue(indent) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let worker { + return try await pool.methodResult(handle: self, name: "to_json", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "to_json", args: __args, kwargs: __kwargs) + } + + public func equals(other: PoolSeries, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "equals", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "equals", args: __args, kwargs: __kwargs) + } + + public func compare(other: PoolSeries, align_axis: Int? = nil, keep_shape: Bool? = nil, keep_equal: Bool? = nil, result_names: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let align_axis = align_axis { + __kwargs["align_axis"] = RemotePythonValue(align_axis) + } + if let keep_shape = keep_shape { + __kwargs["keep_shape"] = RemotePythonValue(keep_shape) + } + if let keep_equal = keep_equal { + __kwargs["keep_equal"] = RemotePythonValue(keep_equal) + } + if let result_names = result_names { + __kwargs["result_names"] = RemotePythonValue(result_names) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "compare", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "compare", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } +} + + +public struct PoolDataFrameGroupBy: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func agg(`func`: String, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func aggregate(`func`: String, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func sum(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func mean(numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func median(numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func min(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func max(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func std(ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func `var`(ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func count(worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "count", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "count", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func nunique(dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nunique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nunique", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func first(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "first", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "first", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func last(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "last", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "last", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func prod(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func size(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "size", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "size", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cumsum(axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cumprod(axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cummin(axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cummax(axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func cumcount(ascending: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumcount", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumcount", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func rank(method: String? = nil, ascending: Bool? = nil, na_option: String? = nil, pct: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let na_option = na_option { + __kwargs["na_option"] = RemotePythonValue(na_option) + } + if let pct = pct { + __kwargs["pct"] = RemotePythonValue(pct) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func quantile(q: Double? = nil, interpolation: String? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let q = q { + __kwargs["q"] = RemotePythonValue(q) + } + if let interpolation = interpolation { + __kwargs["interpolation"] = RemotePythonValue(interpolation) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func describe(percentiles: [Double]? = nil, include: (any PythonConvertible)? = nil, exclude: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let percentiles = percentiles { + __kwargs["percentiles"] = RemotePythonValue(percentiles) + } + if let include = include { + __kwargs["include"] = RemotePythonValue(include) + } + if let exclude = exclude { + __kwargs["exclude"] = RemotePythonValue(exclude) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func apply(`func`: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(`func`: any PythonConvertible, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func filter(`func`: any PythonConvertible, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "filter", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "filter", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func get_group(name: any PythonConvertible, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(name)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_group", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_group", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func head(n: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func tail(n: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func nth(n: Int, dropna: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nth", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nth", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func shift(periods: Int? = nil, freq: String? = nil, axis: Int? = nil, fill_value: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let freq = freq { + __kwargs["freq"] = RemotePythonValue(freq) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "shift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "shift", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func diff(periods: Int? = nil, axis: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func pct_change(periods: Int? = nil, fill_method: String? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let fill_method = fill_method { + __kwargs["fill_method"] = RemotePythonValue(fill_method) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func fillna(value: (any PythonConvertible)? = nil, method: String? = nil, axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func ffill(limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func bfill(limit: Int? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } +} + + +public struct PoolSeriesGroupBy: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func agg(`func`: String, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "agg", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func aggregate(`func`: String, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "aggregate", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func sum(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sum", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func mean(numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "mean", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func median(numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "median", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func min(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func max(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func std(ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "std", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func `var`(ddof: Int? = nil, numeric_only: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ddof = ddof { + __kwargs["ddof"] = RemotePythonValue(ddof) + } + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "var", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func count(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "count", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "count", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func nunique(dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nunique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nunique", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func first(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "first", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "first", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func last(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "last", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "last", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func prod(numeric_only: Bool? = nil, min_count: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let numeric_only = numeric_only { + __kwargs["numeric_only"] = RemotePythonValue(numeric_only) + } + if let min_count = min_count { + __kwargs["min_count"] = RemotePythonValue(min_count) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "prod", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func size(worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "size", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "size", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cumsum(axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumsum", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cumprod(axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumprod", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cummin(axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummin", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cummax(axis: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cummax", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func cumcount(ascending: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cumcount", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cumcount", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func rank(method: String? = nil, ascending: Bool? = nil, na_option: String? = nil, pct: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let na_option = na_option { + __kwargs["na_option"] = RemotePythonValue(na_option) + } + if let pct = pct { + __kwargs["pct"] = RemotePythonValue(pct) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rank", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func quantile(q: Double? = nil, interpolation: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let q = q { + __kwargs["q"] = RemotePythonValue(q) + } + if let interpolation = interpolation { + __kwargs["interpolation"] = RemotePythonValue(interpolation) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "quantile", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func describe(percentiles: [Double]? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let percentiles = percentiles { + __kwargs["percentiles"] = RemotePythonValue(percentiles) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "describe", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func value_counts(normalize: Bool? = nil, sort: Bool? = nil, ascending: Bool? = nil, bins: Int? = nil, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let bins = bins { + __kwargs["bins"] = RemotePythonValue(bins) + } + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "value_counts", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "value_counts", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func unique(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "unique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "unique", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func apply(`func`: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(`func`: any PythonConvertible, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func filter(`func`: any PythonConvertible, dropna: Bool? = nil, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(`func`)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "filter", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "filter", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func get_group(name: any PythonConvertible, worker: Int? = nil) async throws -> PoolSeries { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(name)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_group", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_group", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func head(n: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "head", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func tail(n: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n = n { + __kwargs["n"] = RemotePythonValue(n) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tail", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func nth(n: Int, dropna: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n)) + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "nth", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "nth", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func shift(periods: Int? = nil, freq: String? = nil, axis: Int? = nil, fill_value: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let freq = freq { + __kwargs["freq"] = RemotePythonValue(freq) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let fill_value = fill_value { + __kwargs["fill_value"] = RemotePythonValue(fill_value) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "shift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "shift", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func diff(periods: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "diff", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func pct_change(periods: Int? = nil, fill_method: String? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let periods = periods { + __kwargs["periods"] = RemotePythonValue(periods) + } + if let fill_method = fill_method { + __kwargs["fill_method"] = RemotePythonValue(fill_method) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "pct_change", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func fillna(value: (any PythonConvertible)? = nil, method: String? = nil, axis: Int? = nil, inplace: Bool? = nil, limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let inplace = inplace { + __kwargs["inplace"] = RemotePythonValue(inplace) + } + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func ffill(limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "ffill", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func bfill(limit: Int? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let limit = limit { + __kwargs["limit"] = RemotePythonValue(limit) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "bfill", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } +} + + +public struct PoolIndex: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func getItem(key: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(key)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "__getitem__", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "__getitem__", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func tolist(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "tolist", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "tolist", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_list(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_list", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_list", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_numpy(dtype: String? = nil, copy: Bool? = nil, na_value: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let na_value = na_value { + __kwargs["na_value"] = RemotePythonValue(na_value) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_numpy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_numpy", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func to_series(index: (any PythonConvertible)? = nil, name: String? = nil, worker: Int? = nil) async throws -> PoolSeries { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_series", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_series", args: __args, kwargs: __kwargs) + } + return PoolSeries(pool: pool, owned: __owned) + } + + public func to_frame(index: Bool? = nil, name: String? = nil, worker: Int? = nil) async throws -> PoolDataFrame { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let index = index { + __kwargs["index"] = RemotePythonValue(index) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "to_frame", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "to_frame", args: __args, kwargs: __kwargs) + } + return PoolDataFrame(pool: pool, owned: __owned) + } + + public func astype(dtype: String, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(dtype)) + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "astype", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func copy(name: String? = nil, deep: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "copy", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func unique(level: Int? = nil, worker: Int? = nil) async throws -> PoolIndex { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "unique", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "unique", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func nunique(dropna: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let dropna = dropna { + __kwargs["dropna"] = RemotePythonValue(dropna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "nunique", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "nunique", args: __args, kwargs: __kwargs) + } + + public func duplicated(keep: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "duplicated", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "duplicated", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func drop_duplicates(keep: String? = nil, worker: Int? = nil) async throws -> PoolIndex { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let keep = keep { + __kwargs["keep"] = RemotePythonValue(keep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "drop_duplicates", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "drop_duplicates", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func dropna(how: String? = nil, worker: Int? = nil) async throws -> PoolIndex { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let how = how { + __kwargs["how"] = RemotePythonValue(how) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "dropna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "dropna", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func fillna(value: any PythonConvertible, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(value)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fillna", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func isna(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isna", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func isnull(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isnull", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func notna(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "notna", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "notna", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func notnull(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "notnull", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "notnull", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_loc(key: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(key)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_loc", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_loc", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_indexer(target: any PythonConvertible, method: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(target)) + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_indexer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_indexer", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func isin(values: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(values)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "isin", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "isin", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func sort_values(ascending: Bool? = nil, na_position: String? = nil, worker: Int? = nil) async throws -> PoolIndex { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + if let na_position = na_position { + __kwargs["na_position"] = RemotePythonValue(na_position) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "sort_values", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "sort_values", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func argsort(ascending: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ascending = ascending { + __kwargs["ascending"] = RemotePythonValue(ascending) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "argsort", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "argsort", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func min(skipna: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "min", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func max(skipna: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "max", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func argmin(skipna: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "argmin", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "argmin", args: __args, kwargs: __kwargs) + } + + public func argmax(skipna: Bool? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let skipna = skipna { + __kwargs["skipna"] = RemotePythonValue(skipna) + } + if let worker { + return try await pool.methodResult(handle: self, name: "argmax", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "argmax", args: __args, kwargs: __kwargs) + } + + public func append(other: any PythonConvertible, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "append", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "append", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func insert(loc: Int, item: any PythonConvertible, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(loc)) + __args.append(RemotePythonValue(item)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "insert", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "insert", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func delete(loc: any PythonConvertible, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(loc)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "delete", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "delete", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func drop(labels: any PythonConvertible, errors: String? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels)) + var __kwargs: [String: RemotePythonValue] = [:] + if let errors = errors { + __kwargs["errors"] = RemotePythonValue(errors) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "drop", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "drop", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func union(other: PoolIndex, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "union", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "union", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func intersection(other: PoolIndex, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "intersection", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "intersection", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func difference(other: PoolIndex, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "difference", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "difference", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func symmetric_difference(other: PoolIndex, sort: Bool? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sort = sort { + __kwargs["sort"] = RemotePythonValue(sort) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "symmetric_difference", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "symmetric_difference", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func rename(name: String, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(name)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "rename", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "rename", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func set_names(names: any PythonConvertible, level: Int? = nil, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(names)) + var __kwargs: [String: RemotePythonValue] = [:] + if let level = level { + __kwargs["level"] = RemotePythonValue(level) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "set_names", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "set_names", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func map(mapper: any PythonConvertible, worker: Int? = nil) async throws -> PoolIndex { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(mapper)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "map", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "map", args: __args, kwargs: __kwargs) + } + return PoolIndex(pool: pool, owned: __owned) + } + + public func equals(other: PoolIndex, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "equals", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "equals", args: __args, kwargs: __kwargs) + } + + public func identical(other: PoolIndex, worker: Int? = nil) async throws -> Bool { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(other.handle)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "identical", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "identical", args: __args, kwargs: __kwargs) + } +} + +// MARK: - Pandas.Testing + +extension PoolPandas { + /// Pool-bound `pandas.testing` namespace. Calls run on this pool. + public var testing: PoolPandasTesting { PoolPandasTesting(pool: pool) } +} + +public struct PoolPandasTesting: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func assert_frame_equal(left: PoolDataFrame, right: PoolDataFrame, check_dtype: Bool? = nil, check_index_type: Bool? = nil, check_column_type: Bool? = nil, check_frame_type: Bool? = nil, check_names: Bool? = nil, by_blocks: Bool? = nil, check_exact: Bool? = nil, check_datetimelike_compat: Bool? = nil, check_categorical: Bool? = nil, check_like: Bool? = nil, check_freq: Bool? = nil, check_flags: Bool? = nil, rtol: Double? = nil, atol: Double? = nil, obj: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left.handle)) + __args.append(RemotePythonValue(right.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let check_dtype = check_dtype { + __kwargs["check_dtype"] = RemotePythonValue(check_dtype) + } + if let check_index_type = check_index_type { + __kwargs["check_index_type"] = RemotePythonValue(check_index_type) + } + if let check_column_type = check_column_type { + __kwargs["check_column_type"] = RemotePythonValue(check_column_type) + } + if let check_frame_type = check_frame_type { + __kwargs["check_frame_type"] = RemotePythonValue(check_frame_type) + } + if let check_names = check_names { + __kwargs["check_names"] = RemotePythonValue(check_names) + } + if let by_blocks = by_blocks { + __kwargs["by_blocks"] = RemotePythonValue(by_blocks) + } + if let check_exact = check_exact { + __kwargs["check_exact"] = RemotePythonValue(check_exact) + } + if let check_datetimelike_compat = check_datetimelike_compat { + __kwargs["check_datetimelike_compat"] = RemotePythonValue(check_datetimelike_compat) + } + if let check_categorical = check_categorical { + __kwargs["check_categorical"] = RemotePythonValue(check_categorical) + } + if let check_like = check_like { + __kwargs["check_like"] = RemotePythonValue(check_like) + } + if let check_freq = check_freq { + __kwargs["check_freq"] = RemotePythonValue(check_freq) + } + if let check_flags = check_flags { + __kwargs["check_flags"] = RemotePythonValue(check_flags) + } + if let rtol = rtol { + __kwargs["rtol"] = RemotePythonValue(rtol) + } + if let atol = atol { + __kwargs["atol"] = RemotePythonValue(atol) + } + if let obj = obj { + __kwargs["obj"] = RemotePythonValue(obj) + } + if let worker { + try await pool.invokeVoid(module: "pandas.testing", function: "assert_frame_equal", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "pandas.testing", function: "assert_frame_equal", args: __args, kwargs: __kwargs) + } + + public func assert_series_equal(left: PoolSeries, right: PoolSeries, check_dtype: Bool? = nil, check_index_type: Bool? = nil, check_series_type: Bool? = nil, check_names: Bool? = nil, check_exact: Bool? = nil, check_datetimelike_compat: Bool? = nil, check_categorical: Bool? = nil, check_category_order: Bool? = nil, check_freq: Bool? = nil, check_flags: Bool? = nil, rtol: Double? = nil, atol: Double? = nil, obj: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left.handle)) + __args.append(RemotePythonValue(right.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let check_dtype = check_dtype { + __kwargs["check_dtype"] = RemotePythonValue(check_dtype) + } + if let check_index_type = check_index_type { + __kwargs["check_index_type"] = RemotePythonValue(check_index_type) + } + if let check_series_type = check_series_type { + __kwargs["check_series_type"] = RemotePythonValue(check_series_type) + } + if let check_names = check_names { + __kwargs["check_names"] = RemotePythonValue(check_names) + } + if let check_exact = check_exact { + __kwargs["check_exact"] = RemotePythonValue(check_exact) + } + if let check_datetimelike_compat = check_datetimelike_compat { + __kwargs["check_datetimelike_compat"] = RemotePythonValue(check_datetimelike_compat) + } + if let check_categorical = check_categorical { + __kwargs["check_categorical"] = RemotePythonValue(check_categorical) + } + if let check_category_order = check_category_order { + __kwargs["check_category_order"] = RemotePythonValue(check_category_order) + } + if let check_freq = check_freq { + __kwargs["check_freq"] = RemotePythonValue(check_freq) + } + if let check_flags = check_flags { + __kwargs["check_flags"] = RemotePythonValue(check_flags) + } + if let rtol = rtol { + __kwargs["rtol"] = RemotePythonValue(rtol) + } + if let atol = atol { + __kwargs["atol"] = RemotePythonValue(atol) + } + if let obj = obj { + __kwargs["obj"] = RemotePythonValue(obj) + } + if let worker { + try await pool.invokeVoid(module: "pandas.testing", function: "assert_series_equal", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "pandas.testing", function: "assert_series_equal", args: __args, kwargs: __kwargs) + } + + public func assert_index_equal(left: PoolIndex, right: PoolIndex, exact: Bool? = nil, check_names: Bool? = nil, check_exact: Bool? = nil, check_categorical: Bool? = nil, check_order: Bool? = nil, rtol: Double? = nil, atol: Double? = nil, obj: String? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(left.handle)) + __args.append(RemotePythonValue(right.handle)) + var __kwargs: [String: RemotePythonValue] = [:] + if let exact = exact { + __kwargs["exact"] = RemotePythonValue(exact) + } + if let check_names = check_names { + __kwargs["check_names"] = RemotePythonValue(check_names) + } + if let check_exact = check_exact { + __kwargs["check_exact"] = RemotePythonValue(check_exact) + } + if let check_categorical = check_categorical { + __kwargs["check_categorical"] = RemotePythonValue(check_categorical) + } + if let check_order = check_order { + __kwargs["check_order"] = RemotePythonValue(check_order) + } + if let rtol = rtol { + __kwargs["rtol"] = RemotePythonValue(rtol) + } + if let atol = atol { + __kwargs["atol"] = RemotePythonValue(atol) + } + if let obj = obj { + __kwargs["obj"] = RemotePythonValue(obj) + } + if let worker { + try await pool.invokeVoid(module: "pandas.testing", function: "assert_index_equal", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "pandas.testing", function: "assert_index_equal", args: __args, kwargs: __kwargs) + } +} + diff --git a/Examples/Studio/Sources/SwiftPythonKit/PoolSklearn.swift b/Examples/Studio/Sources/SwiftPythonKit/PoolSklearn.swift new file mode 100644 index 0000000..a8b1706 --- /dev/null +++ b/Examples/Studio/Sources/SwiftPythonKit/PoolSklearn.swift @@ -0,0 +1,19717 @@ +// Generated by swift-python-gen +// Do not edit manually + +import SwiftPythonRuntime +import Foundation + +// MARK: - Sklearn.Cluster + +extension PoolSklearn { + /// Pool-bound `sklearn.cluster` namespace. Calls run on this pool. + public var cluster: PoolSklearnCluster { PoolSklearnCluster(pool: pool) } +} + +public struct PoolSklearnCluster: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func k_means(X: any PythonConvertible, n_clusters: Int, sample_weight: (any PythonConvertible)? = nil, `init`: String? = nil, n_init: String? = nil, max_iter: Int? = nil, verbose: Bool? = nil, tol: Double? = nil, random_state: Int? = nil, copy_x: Bool? = nil, algorithm: String? = nil, return_n_iter: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(n_clusters)) + if let sample_weight = sample_weight { + __args.append(RemotePythonValue(sample_weight)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let copy_x = copy_x { + __kwargs["copy_x"] = RemotePythonValue(copy_x) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "k_means", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "k_means", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kmeans_plusplus(X: any PythonConvertible, n_clusters: Int, x_squared_norms: (any PythonConvertible)? = nil, random_state: Int? = nil, n_local_trials: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(n_clusters)) + if let x_squared_norms = x_squared_norms { + __args.append(RemotePythonValue(x_squared_norms)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_local_trials = n_local_trials { + __kwargs["n_local_trials"] = RemotePythonValue(n_local_trials) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "kmeans_plusplus", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "kmeans_plusplus", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func mean_shift(X: any PythonConvertible, bandwidth: Double? = nil, seeds: (any PythonConvertible)? = nil, bin_seeding: Bool? = nil, min_bin_freq: Int? = nil, cluster_all: Bool? = nil, max_iter: Int? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let bandwidth = bandwidth { + __kwargs["bandwidth"] = RemotePythonValue(bandwidth) + } + if let seeds = seeds { + __kwargs["seeds"] = RemotePythonValue(seeds) + } + if let bin_seeding = bin_seeding { + __kwargs["bin_seeding"] = RemotePythonValue(bin_seeding) + } + if let min_bin_freq = min_bin_freq { + __kwargs["min_bin_freq"] = RemotePythonValue(min_bin_freq) + } + if let cluster_all = cluster_all { + __kwargs["cluster_all"] = RemotePythonValue(cluster_all) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "mean_shift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "mean_shift", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func estimate_bandwidth(X: any PythonConvertible, quantile: Double? = nil, n_samples: Int? = nil, random_state: Int? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let quantile = quantile { + __kwargs["quantile"] = RemotePythonValue(quantile) + } + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.cluster", function: "estimate_bandwidth", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.cluster", function: "estimate_bandwidth", args: __args, kwargs: __kwargs) + } + + public func affinity_propagation(S: any PythonConvertible, preference: Double? = nil, convergence_iter: Int? = nil, max_iter: Int? = nil, damping: Double? = nil, copy: Bool? = nil, verbose: Bool? = nil, return_n_iter: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(S)) + var __kwargs: [String: RemotePythonValue] = [:] + if let preference = preference { + __kwargs["preference"] = RemotePythonValue(preference) + } + if let convergence_iter = convergence_iter { + __kwargs["convergence_iter"] = RemotePythonValue(convergence_iter) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let damping = damping { + __kwargs["damping"] = RemotePythonValue(damping) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "affinity_propagation", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "affinity_propagation", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func dbscan(X: any PythonConvertible, eps: Double? = nil, min_samples: Int? = nil, metric: String? = nil, metric_params: (any PythonConvertible)? = nil, algorithm: String? = nil, leaf_size: Int? = nil, p: Double? = nil, sample_weight: (any PythonConvertible)? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let min_samples = min_samples { + __kwargs["min_samples"] = RemotePythonValue(min_samples) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "dbscan", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "dbscan", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func spectral_clustering(affinity: any PythonConvertible, n_clusters: Int? = nil, n_components: Int? = nil, eigen_solver: String? = nil, random_state: Int? = nil, n_init: Int? = nil, eigen_tol: String? = nil, assign_labels: String? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(affinity)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let eigen_solver = eigen_solver { + __kwargs["eigen_solver"] = RemotePythonValue(eigen_solver) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let eigen_tol = eigen_tol { + __kwargs["eigen_tol"] = RemotePythonValue(eigen_tol) + } + if let assign_labels = assign_labels { + __kwargs["assign_labels"] = RemotePythonValue(assign_labels) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "spectral_clustering", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "spectral_clustering", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func ward_tree(X: any PythonConvertible, connectivity: (any PythonConvertible)? = nil, n_clusters: Int? = nil, return_distance: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let connectivity = connectivity { + __args.append(RemotePythonValue(connectivity)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "ward_tree", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "ward_tree", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cluster_optics_dbscan(reachability: any PythonConvertible, core_distances: any PythonConvertible, ordering: any PythonConvertible, eps: Double, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(reachability)) + __args.append(RemotePythonValue(core_distances)) + __args.append(RemotePythonValue(ordering)) + __args.append(RemotePythonValue(eps)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "cluster_optics_dbscan", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "cluster_optics_dbscan", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cluster_optics_xi(reachability: any PythonConvertible, predecessor: any PythonConvertible, ordering: any PythonConvertible, min_samples: Int, min_cluster_size: Int? = nil, xi: Double? = nil, predecessor_correction: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(reachability)) + __args.append(RemotePythonValue(predecessor)) + __args.append(RemotePythonValue(ordering)) + __args.append(RemotePythonValue(min_samples)) + var __kwargs: [String: RemotePythonValue] = [:] + if let min_cluster_size = min_cluster_size { + __kwargs["min_cluster_size"] = RemotePythonValue(min_cluster_size) + } + if let xi = xi { + __kwargs["xi"] = RemotePythonValue(xi) + } + if let predecessor_correction = predecessor_correction { + __kwargs["predecessor_correction"] = RemotePythonValue(predecessor_correction) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "cluster_optics_xi", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "cluster_optics_xi", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func compute_optics_graph(X: any PythonConvertible, min_samples: Int? = nil, max_eps: Double? = nil, metric: String? = nil, p: Int? = nil, metric_params: (any PythonConvertible)? = nil, algorithm: String? = nil, leaf_size: Int? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let min_samples = min_samples { + __kwargs["min_samples"] = RemotePythonValue(min_samples) + } + if let max_eps = max_eps { + __kwargs["max_eps"] = RemotePythonValue(max_eps) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "compute_optics_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "compute_optics_graph", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func KMeans(n_clusters: Int? = nil, `init`: String? = nil, n_init: String? = nil, max_iter: Int? = nil, tol: Double? = nil, verbose: Int? = nil, random_state: Int? = nil, copy_x: Bool? = nil, algorithm: String? = nil, worker: Int? = nil) async throws -> PoolKMeans { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let copy_x = copy_x { + __kwargs["copy_x"] = RemotePythonValue(copy_x) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "KMeans", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "KMeans", args: __args, kwargs: __kwargs) + } + return PoolKMeans(pool: pool, owned: __owned) + } + + public func MiniBatchKMeans(n_clusters: Int? = nil, `init`: String? = nil, max_iter: Int? = nil, batch_size: Int? = nil, verbose: Int? = nil, compute_labels: Bool? = nil, random_state: Int? = nil, tol: Double? = nil, max_no_improvement: Int? = nil, init_size: Int? = nil, n_init: String? = nil, reassignment_ratio: Double? = nil, worker: Int? = nil) async throws -> PoolMiniBatchKMeans { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let batch_size = batch_size { + __kwargs["batch_size"] = RemotePythonValue(batch_size) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let compute_labels = compute_labels { + __kwargs["compute_labels"] = RemotePythonValue(compute_labels) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let max_no_improvement = max_no_improvement { + __kwargs["max_no_improvement"] = RemotePythonValue(max_no_improvement) + } + if let init_size = init_size { + __kwargs["init_size"] = RemotePythonValue(init_size) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let reassignment_ratio = reassignment_ratio { + __kwargs["reassignment_ratio"] = RemotePythonValue(reassignment_ratio) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "MiniBatchKMeans", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "MiniBatchKMeans", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchKMeans(pool: pool, owned: __owned) + } + + public func DBSCAN(eps: Double? = nil, min_samples: Int? = nil, metric: String? = nil, metric_params: (any PythonConvertible)? = nil, algorithm: String? = nil, leaf_size: Int? = nil, p: Double? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolDBSCAN { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let min_samples = min_samples { + __kwargs["min_samples"] = RemotePythonValue(min_samples) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "DBSCAN", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "DBSCAN", args: __args, kwargs: __kwargs) + } + return PoolDBSCAN(pool: pool, owned: __owned) + } + + public func HDBSCAN(min_cluster_size: Int? = nil, min_samples: Int? = nil, cluster_selection_epsilon: Double? = nil, max_cluster_size: Int? = nil, metric: String? = nil, metric_params: (any PythonConvertible)? = nil, alpha: Double? = nil, algorithm: String? = nil, leaf_size: Int? = nil, n_jobs: Int? = nil, cluster_selection_method: String? = nil, allow_single_cluster: Bool? = nil, store_centers: String? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolHDBSCAN { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let min_cluster_size = min_cluster_size { + __kwargs["min_cluster_size"] = RemotePythonValue(min_cluster_size) + } + if let min_samples = min_samples { + __kwargs["min_samples"] = RemotePythonValue(min_samples) + } + if let cluster_selection_epsilon = cluster_selection_epsilon { + __kwargs["cluster_selection_epsilon"] = RemotePythonValue(cluster_selection_epsilon) + } + if let max_cluster_size = max_cluster_size { + __kwargs["max_cluster_size"] = RemotePythonValue(max_cluster_size) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let cluster_selection_method = cluster_selection_method { + __kwargs["cluster_selection_method"] = RemotePythonValue(cluster_selection_method) + } + if let allow_single_cluster = allow_single_cluster { + __kwargs["allow_single_cluster"] = RemotePythonValue(allow_single_cluster) + } + if let store_centers = store_centers { + __kwargs["store_centers"] = RemotePythonValue(store_centers) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "HDBSCAN", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "HDBSCAN", args: __args, kwargs: __kwargs) + } + return PoolHDBSCAN(pool: pool, owned: __owned) + } + + public func OPTICS(min_samples: Int? = nil, max_eps: Double? = nil, metric: String? = nil, p: Int? = nil, metric_params: (any PythonConvertible)? = nil, cluster_method: String? = nil, eps: Double? = nil, xi: Double? = nil, predecessor_correction: Bool? = nil, min_cluster_size: Int? = nil, algorithm: String? = nil, leaf_size: Int? = nil, memory: String? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolOPTICS { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let min_samples = min_samples { + __kwargs["min_samples"] = RemotePythonValue(min_samples) + } + if let max_eps = max_eps { + __kwargs["max_eps"] = RemotePythonValue(max_eps) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let cluster_method = cluster_method { + __kwargs["cluster_method"] = RemotePythonValue(cluster_method) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let xi = xi { + __kwargs["xi"] = RemotePythonValue(xi) + } + if let predecessor_correction = predecessor_correction { + __kwargs["predecessor_correction"] = RemotePythonValue(predecessor_correction) + } + if let min_cluster_size = min_cluster_size { + __kwargs["min_cluster_size"] = RemotePythonValue(min_cluster_size) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let memory = memory { + __kwargs["memory"] = RemotePythonValue(memory) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "OPTICS", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "OPTICS", args: __args, kwargs: __kwargs) + } + return PoolOPTICS(pool: pool, owned: __owned) + } + + public func AgglomerativeClustering(n_clusters: Int? = nil, metric: String? = nil, memory: String? = nil, connectivity: (any PythonConvertible)? = nil, compute_full_tree: String? = nil, linkage: String? = nil, distance_threshold: Double? = nil, compute_distances: Bool? = nil, worker: Int? = nil) async throws -> PoolAgglomerativeClustering { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let memory = memory { + __kwargs["memory"] = RemotePythonValue(memory) + } + if let connectivity = connectivity { + __kwargs["connectivity"] = RemotePythonValue(connectivity) + } + if let compute_full_tree = compute_full_tree { + __kwargs["compute_full_tree"] = RemotePythonValue(compute_full_tree) + } + if let linkage = linkage { + __kwargs["linkage"] = RemotePythonValue(linkage) + } + if let distance_threshold = distance_threshold { + __kwargs["distance_threshold"] = RemotePythonValue(distance_threshold) + } + if let compute_distances = compute_distances { + __kwargs["compute_distances"] = RemotePythonValue(compute_distances) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "AgglomerativeClustering", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "AgglomerativeClustering", args: __args, kwargs: __kwargs) + } + return PoolAgglomerativeClustering(pool: pool, owned: __owned) + } + + public func FeatureAgglomeration(n_clusters: Int? = nil, metric: String? = nil, memory: String? = nil, connectivity: (any PythonConvertible)? = nil, compute_full_tree: String? = nil, linkage: String? = nil, pooling_func: (any PythonConvertible)? = nil, distance_threshold: Double? = nil, compute_distances: Bool? = nil, worker: Int? = nil) async throws -> PoolFeatureAgglomeration { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let memory = memory { + __kwargs["memory"] = RemotePythonValue(memory) + } + if let connectivity = connectivity { + __kwargs["connectivity"] = RemotePythonValue(connectivity) + } + if let compute_full_tree = compute_full_tree { + __kwargs["compute_full_tree"] = RemotePythonValue(compute_full_tree) + } + if let linkage = linkage { + __kwargs["linkage"] = RemotePythonValue(linkage) + } + if let pooling_func = pooling_func { + __kwargs["pooling_func"] = RemotePythonValue(pooling_func) + } + if let distance_threshold = distance_threshold { + __kwargs["distance_threshold"] = RemotePythonValue(distance_threshold) + } + if let compute_distances = compute_distances { + __kwargs["compute_distances"] = RemotePythonValue(compute_distances) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "FeatureAgglomeration", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "FeatureAgglomeration", args: __args, kwargs: __kwargs) + } + return PoolFeatureAgglomeration(pool: pool, owned: __owned) + } + + public func SpectralClustering(n_clusters: Int? = nil, eigen_solver: String? = nil, n_components: Int? = nil, random_state: Int? = nil, n_init: Int? = nil, gamma: Double? = nil, affinity: String? = nil, n_neighbors: Int? = nil, eigen_tol: String? = nil, assign_labels: String? = nil, degree: Double? = nil, coef0: Double? = nil, kernel_params: (any PythonConvertible)? = nil, n_jobs: Int? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolSpectralClustering { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let eigen_solver = eigen_solver { + __kwargs["eigen_solver"] = RemotePythonValue(eigen_solver) + } + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let affinity = affinity { + __kwargs["affinity"] = RemotePythonValue(affinity) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let eigen_tol = eigen_tol { + __kwargs["eigen_tol"] = RemotePythonValue(eigen_tol) + } + if let assign_labels = assign_labels { + __kwargs["assign_labels"] = RemotePythonValue(assign_labels) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let kernel_params = kernel_params { + __kwargs["kernel_params"] = RemotePythonValue(kernel_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "SpectralClustering", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "SpectralClustering", args: __args, kwargs: __kwargs) + } + return PoolSpectralClustering(pool: pool, owned: __owned) + } + + public func SpectralBiclustering(n_clusters: Int? = nil, method: String? = nil, n_components: Int? = nil, n_best: Int? = nil, svd_method: String? = nil, n_svd_vecs: Int? = nil, mini_batch: Bool? = nil, `init`: String? = nil, n_init: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolSpectralBiclustering { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let n_best = n_best { + __kwargs["n_best"] = RemotePythonValue(n_best) + } + if let svd_method = svd_method { + __kwargs["svd_method"] = RemotePythonValue(svd_method) + } + if let n_svd_vecs = n_svd_vecs { + __kwargs["n_svd_vecs"] = RemotePythonValue(n_svd_vecs) + } + if let mini_batch = mini_batch { + __kwargs["mini_batch"] = RemotePythonValue(mini_batch) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "SpectralBiclustering", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "SpectralBiclustering", args: __args, kwargs: __kwargs) + } + return PoolSpectralBiclustering(pool: pool, owned: __owned) + } + + public func SpectralCoclustering(n_clusters: Int? = nil, svd_method: String? = nil, n_svd_vecs: Int? = nil, mini_batch: Bool? = nil, `init`: String? = nil, n_init: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolSpectralCoclustering { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let svd_method = svd_method { + __kwargs["svd_method"] = RemotePythonValue(svd_method) + } + if let n_svd_vecs = n_svd_vecs { + __kwargs["n_svd_vecs"] = RemotePythonValue(n_svd_vecs) + } + if let mini_batch = mini_batch { + __kwargs["mini_batch"] = RemotePythonValue(mini_batch) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "SpectralCoclustering", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "SpectralCoclustering", args: __args, kwargs: __kwargs) + } + return PoolSpectralCoclustering(pool: pool, owned: __owned) + } + + public func Birch(threshold: Double? = nil, branching_factor: Int? = nil, n_clusters: Int? = nil, compute_labels: Bool? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolBirch { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let threshold = threshold { + __kwargs["threshold"] = RemotePythonValue(threshold) + } + if let branching_factor = branching_factor { + __kwargs["branching_factor"] = RemotePythonValue(branching_factor) + } + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let compute_labels = compute_labels { + __kwargs["compute_labels"] = RemotePythonValue(compute_labels) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "Birch", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "Birch", args: __args, kwargs: __kwargs) + } + return PoolBirch(pool: pool, owned: __owned) + } + + public func MeanShift(bandwidth: Double? = nil, seeds: (any PythonConvertible)? = nil, bin_seeding: Bool? = nil, min_bin_freq: Int? = nil, cluster_all: Bool? = nil, n_jobs: Int? = nil, max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolMeanShift { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let bandwidth = bandwidth { + __kwargs["bandwidth"] = RemotePythonValue(bandwidth) + } + if let seeds = seeds { + __kwargs["seeds"] = RemotePythonValue(seeds) + } + if let bin_seeding = bin_seeding { + __kwargs["bin_seeding"] = RemotePythonValue(bin_seeding) + } + if let min_bin_freq = min_bin_freq { + __kwargs["min_bin_freq"] = RemotePythonValue(min_bin_freq) + } + if let cluster_all = cluster_all { + __kwargs["cluster_all"] = RemotePythonValue(cluster_all) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "MeanShift", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "MeanShift", args: __args, kwargs: __kwargs) + } + return PoolMeanShift(pool: pool, owned: __owned) + } + + public func AffinityPropagation(damping: Double? = nil, max_iter: Int? = nil, convergence_iter: Int? = nil, copy: Bool? = nil, preference: Double? = nil, affinity: String? = nil, verbose: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolAffinityPropagation { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let damping = damping { + __kwargs["damping"] = RemotePythonValue(damping) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let convergence_iter = convergence_iter { + __kwargs["convergence_iter"] = RemotePythonValue(convergence_iter) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let preference = preference { + __kwargs["preference"] = RemotePythonValue(preference) + } + if let affinity = affinity { + __kwargs["affinity"] = RemotePythonValue(affinity) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "AffinityPropagation", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "AffinityPropagation", args: __args, kwargs: __kwargs) + } + return PoolAffinityPropagation(pool: pool, owned: __owned) + } + + public func BisectingKMeans(n_clusters: Int? = nil, `init`: String? = nil, n_init: Int? = nil, random_state: Int? = nil, max_iter: Int? = nil, verbose: Int? = nil, tol: Double? = nil, copy_x: Bool? = nil, algorithm: String? = nil, bisecting_strategy: String? = nil, worker: Int? = nil) async throws -> PoolBisectingKMeans { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_clusters = n_clusters { + __kwargs["n_clusters"] = RemotePythonValue(n_clusters) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let n_init = n_init { + __kwargs["n_init"] = RemotePythonValue(n_init) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let copy_x = copy_x { + __kwargs["copy_x"] = RemotePythonValue(copy_x) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let bisecting_strategy = bisecting_strategy { + __kwargs["bisecting_strategy"] = RemotePythonValue(bisecting_strategy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "BisectingKMeans", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.cluster", function: "BisectingKMeans", args: __args, kwargs: __kwargs) + } + return PoolBisectingKMeans(pool: pool, owned: __owned) + } +} + + +public struct PoolKMeans: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolKMeans { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolKMeans(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolMiniBatchKMeans: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMiniBatchKMeans { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchKMeans(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMiniBatchKMeans { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchKMeans(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolDBSCAN: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolDBSCAN { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolDBSCAN(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolHDBSCAN: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolHDBSCAN { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolHDBSCAN(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func dbscan_clustering(cut_distance: Double, min_cluster_size: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(cut_distance)) + var __kwargs: [String: RemotePythonValue] = [:] + if let min_cluster_size = min_cluster_size { + __kwargs["min_cluster_size"] = RemotePythonValue(min_cluster_size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "dbscan_clustering", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "dbscan_clustering", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolOPTICS: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolOPTICS { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolOPTICS(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolAgglomerativeClustering: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolAgglomerativeClustering { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolAgglomerativeClustering(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolFeatureAgglomeration: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolFeatureAgglomeration { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolFeatureAgglomeration(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolSpectralClustering: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSpectralClustering { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSpectralClustering(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolSpectralBiclustering: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSpectralBiclustering { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSpectralBiclustering(pool: pool, owned: __owned) + } + + public func get_indices(i: Int, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(i)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_indices", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_indices", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_shape(i: Int, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(i)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_shape", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_shape", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_submatrix(i: Int, data: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(i)) + __args.append(RemotePythonValue(data)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_submatrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_submatrix", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolSpectralCoclustering: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSpectralCoclustering { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSpectralCoclustering(pool: pool, owned: __owned) + } + + public func get_indices(i: Int, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(i)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_indices", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_indices", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_shape(i: Int, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(i)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_shape", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_shape", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_submatrix(i: Int, data: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(i)) + __args.append(RemotePythonValue(data)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_submatrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_submatrix", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolBirch: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolBirch { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBirch(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolBirch { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolBirch(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolMeanShift: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMeanShift { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMeanShift(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolAffinityPropagation: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolAffinityPropagation { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolAffinityPropagation(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolBisectingKMeans: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolBisectingKMeans { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBisectingKMeans(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + +// MARK: - Sklearn.Datasets + +extension PoolSklearn { + /// Pool-bound `sklearn.datasets` namespace. Calls run on this pool. + public var datasets: PoolSklearnDatasets { PoolSklearnDatasets(pool: pool) } +} + +public struct PoolSklearnDatasets: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func load_iris(return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_iris", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_iris", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func load_digits(n_class: Int? = nil, return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_class = n_class { + __kwargs["n_class"] = RemotePythonValue(n_class) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_digits", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_digits", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func load_wine(return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_wine", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_wine", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func load_breast_cancer(return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_breast_cancer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_breast_cancer", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func load_diabetes(return_X_y: Bool? = nil, as_frame: Bool? = nil, scaled: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + if let scaled = scaled { + __kwargs["scaled"] = RemotePythonValue(scaled) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_diabetes", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_diabetes", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func load_linnerud(return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_linnerud", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_linnerud", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_classification(n_samples: Int? = nil, n_features: Int? = nil, n_informative: Int? = nil, n_redundant: Int? = nil, n_repeated: Int? = nil, n_classes: Int? = nil, n_clusters_per_class: Int? = nil, weights: (any PythonConvertible)? = nil, flip_y: Double? = nil, class_sep: Double? = nil, hypercube: Bool? = nil, shift: Double? = nil, scale: Double? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let n_informative = n_informative { + __kwargs["n_informative"] = RemotePythonValue(n_informative) + } + if let n_redundant = n_redundant { + __kwargs["n_redundant"] = RemotePythonValue(n_redundant) + } + if let n_repeated = n_repeated { + __kwargs["n_repeated"] = RemotePythonValue(n_repeated) + } + if let n_classes = n_classes { + __kwargs["n_classes"] = RemotePythonValue(n_classes) + } + if let n_clusters_per_class = n_clusters_per_class { + __kwargs["n_clusters_per_class"] = RemotePythonValue(n_clusters_per_class) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let flip_y = flip_y { + __kwargs["flip_y"] = RemotePythonValue(flip_y) + } + if let class_sep = class_sep { + __kwargs["class_sep"] = RemotePythonValue(class_sep) + } + if let hypercube = hypercube { + __kwargs["hypercube"] = RemotePythonValue(hypercube) + } + if let shift = shift { + __kwargs["shift"] = RemotePythonValue(shift) + } + if let scale = scale { + __kwargs["scale"] = RemotePythonValue(scale) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_classification", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_classification", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_regression(n_samples: Int? = nil, n_features: Int? = nil, n_informative: Int? = nil, n_targets: Int? = nil, bias: Double? = nil, effective_rank: Int? = nil, tail_strength: Double? = nil, noise: Double? = nil, shuffle: Bool? = nil, coef: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let n_informative = n_informative { + __kwargs["n_informative"] = RemotePythonValue(n_informative) + } + if let n_targets = n_targets { + __kwargs["n_targets"] = RemotePythonValue(n_targets) + } + if let bias = bias { + __kwargs["bias"] = RemotePythonValue(bias) + } + if let effective_rank = effective_rank { + __kwargs["effective_rank"] = RemotePythonValue(effective_rank) + } + if let tail_strength = tail_strength { + __kwargs["tail_strength"] = RemotePythonValue(tail_strength) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let coef = coef { + __kwargs["coef"] = RemotePythonValue(coef) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_regression", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_regression", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_blobs(n_samples: Int? = nil, n_features: Int? = nil, centers: Int? = nil, cluster_std: Double? = nil, center_box: PyHandle? = nil, shuffle: Bool? = nil, random_state: Int? = nil, return_centers: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let centers = centers { + __kwargs["centers"] = RemotePythonValue(centers) + } + if let cluster_std = cluster_std { + __kwargs["cluster_std"] = RemotePythonValue(cluster_std) + } + if let center_box = center_box { + __kwargs["center_box"] = RemotePythonValue(center_box) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let return_centers = return_centers { + __kwargs["return_centers"] = RemotePythonValue(return_centers) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_blobs", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_blobs", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_circles(n_samples: Int? = nil, shuffle: Bool? = nil, noise: Double? = nil, random_state: Int? = nil, factor: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let factor = factor { + __kwargs["factor"] = RemotePythonValue(factor) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_circles", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_circles", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_moons(n_samples: Int? = nil, shuffle: Bool? = nil, noise: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_moons", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_moons", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_gaussian_quantiles(mean: (any PythonConvertible)? = nil, cov: Double? = nil, n_samples: Int? = nil, n_features: Int? = nil, n_classes: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + if let mean = mean { + __args.append(RemotePythonValue(mean)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let cov = cov { + __kwargs["cov"] = RemotePythonValue(cov) + } + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let n_classes = n_classes { + __kwargs["n_classes"] = RemotePythonValue(n_classes) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_gaussian_quantiles", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_gaussian_quantiles", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_hastie_10_2(n_samples: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_hastie_10_2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_hastie_10_2", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_s_curve(n_samples: Int? = nil, noise: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_s_curve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_s_curve", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_swiss_roll(n_samples: Int? = nil, noise: Double? = nil, random_state: Int? = nil, hole: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let hole = hole { + __kwargs["hole"] = RemotePythonValue(hole) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_swiss_roll", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_swiss_roll", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_friedman1(n_samples: Int? = nil, n_features: Int? = nil, noise: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_friedman1", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_friedman1", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_friedman2(n_samples: Int? = nil, noise: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_friedman2", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_friedman2", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_friedman3(n_samples: Int? = nil, noise: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_friedman3", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_friedman3", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_low_rank_matrix(n_samples: Int? = nil, n_features: Int? = nil, effective_rank: Int? = nil, tail_strength: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let effective_rank = effective_rank { + __kwargs["effective_rank"] = RemotePythonValue(effective_rank) + } + if let tail_strength = tail_strength { + __kwargs["tail_strength"] = RemotePythonValue(tail_strength) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_low_rank_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_low_rank_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_sparse_coded_signal(n_samples: Int, n_components: Int, n_features: Int, n_nonzero_coefs: Int, random_state: Int? = nil, data_transposed: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n_samples)) + __args.append(RemotePythonValue(n_components)) + __args.append(RemotePythonValue(n_features)) + __args.append(RemotePythonValue(n_nonzero_coefs)) + var __kwargs: [String: RemotePythonValue] = [:] + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let data_transposed = data_transposed { + __kwargs["data_transposed"] = RemotePythonValue(data_transposed) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_sparse_coded_signal", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_sparse_coded_signal", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_sparse_uncorrelated(n_samples: Int? = nil, n_features: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_sparse_uncorrelated", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_sparse_uncorrelated", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_spd_matrix(n_dim: Int, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(n_dim)) + var __kwargs: [String: RemotePythonValue] = [:] + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_spd_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_spd_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_sparse_spd_matrix(n_dim: Int? = nil, alpha: Double? = nil, norm_diag: Bool? = nil, smallest_coef: Double? = nil, largest_coef: Double? = nil, sparse_format: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_dim = n_dim { + __kwargs["n_dim"] = RemotePythonValue(n_dim) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let norm_diag = norm_diag { + __kwargs["norm_diag"] = RemotePythonValue(norm_diag) + } + if let smallest_coef = smallest_coef { + __kwargs["smallest_coef"] = RemotePythonValue(smallest_coef) + } + if let largest_coef = largest_coef { + __kwargs["largest_coef"] = RemotePythonValue(largest_coef) + } + if let sparse_format = sparse_format { + __kwargs["sparse_format"] = RemotePythonValue(sparse_format) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_sparse_spd_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_sparse_spd_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_biclusters(shape: PyHandle, n_clusters: Int, noise: Double? = nil, minval: Int? = nil, maxval: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + __args.append(RemotePythonValue(n_clusters)) + var __kwargs: [String: RemotePythonValue] = [:] + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let minval = minval { + __kwargs["minval"] = RemotePythonValue(minval) + } + if let maxval = maxval { + __kwargs["maxval"] = RemotePythonValue(maxval) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_biclusters", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_biclusters", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_checkerboard(shape: PyHandle, n_clusters: PyHandle, noise: Double? = nil, minval: Int? = nil, maxval: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(shape)) + __args.append(RemotePythonValue(n_clusters)) + var __kwargs: [String: RemotePythonValue] = [:] + if let noise = noise { + __kwargs["noise"] = RemotePythonValue(noise) + } + if let minval = minval { + __kwargs["minval"] = RemotePythonValue(minval) + } + if let maxval = maxval { + __kwargs["maxval"] = RemotePythonValue(maxval) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_checkerboard", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_checkerboard", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func make_multilabel_classification(n_samples: Int? = nil, n_features: Int? = nil, n_classes: Int? = nil, n_labels: Int? = nil, length: Int? = nil, allow_unlabeled: Bool? = nil, sparse: Bool? = nil, return_indicator: String? = nil, return_distributions: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_samples = n_samples { + __kwargs["n_samples"] = RemotePythonValue(n_samples) + } + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let n_classes = n_classes { + __kwargs["n_classes"] = RemotePythonValue(n_classes) + } + if let n_labels = n_labels { + __kwargs["n_labels"] = RemotePythonValue(n_labels) + } + if let length = length { + __kwargs["length"] = RemotePythonValue(length) + } + if let allow_unlabeled = allow_unlabeled { + __kwargs["allow_unlabeled"] = RemotePythonValue(allow_unlabeled) + } + if let sparse = sparse { + __kwargs["sparse"] = RemotePythonValue(sparse) + } + if let return_indicator = return_indicator { + __kwargs["return_indicator"] = RemotePythonValue(return_indicator) + } + if let return_distributions = return_distributions { + __kwargs["return_distributions"] = RemotePythonValue(return_distributions) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_multilabel_classification", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "make_multilabel_classification", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_olivetti_faces(data_home: String? = nil, shuffle: Bool? = nil, random_state: Int? = nil, download_if_missing: Bool? = nil, return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_olivetti_faces", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_olivetti_faces", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_20newsgroups(data_home: String? = nil, subset: String? = nil, categories: (any PythonConvertible)? = nil, shuffle: Bool? = nil, random_state: Int? = nil, remove: (any PythonConvertible)? = nil, download_if_missing: Bool? = nil, return_X_y: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let categories = categories { + __kwargs["categories"] = RemotePythonValue(categories) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let remove = remove { + __kwargs["remove"] = RemotePythonValue(remove) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_20newsgroups", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_20newsgroups", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_20newsgroups_vectorized(subset: String? = nil, remove: (any PythonConvertible)? = nil, data_home: String? = nil, download_if_missing: Bool? = nil, return_X_y: Bool? = nil, normalize: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let remove = remove { + __kwargs["remove"] = RemotePythonValue(remove) + } + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_20newsgroups_vectorized", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_20newsgroups_vectorized", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_lfw_people(data_home: String? = nil, funneled: Bool? = nil, resize: Double? = nil, min_faces_per_person: Int? = nil, color: Bool? = nil, slice_: (any PythonConvertible)? = nil, download_if_missing: Bool? = nil, return_X_y: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let funneled = funneled { + __kwargs["funneled"] = RemotePythonValue(funneled) + } + if let resize = resize { + __kwargs["resize"] = RemotePythonValue(resize) + } + if let min_faces_per_person = min_faces_per_person { + __kwargs["min_faces_per_person"] = RemotePythonValue(min_faces_per_person) + } + if let color = color { + __kwargs["color"] = RemotePythonValue(color) + } + if let slice_ = slice_ { + __kwargs["slice_"] = RemotePythonValue(slice_) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_lfw_people", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_lfw_people", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_lfw_pairs(subset: String? = nil, data_home: String? = nil, funneled: Bool? = nil, resize: Double? = nil, color: Bool? = nil, slice_: (any PythonConvertible)? = nil, download_if_missing: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let funneled = funneled { + __kwargs["funneled"] = RemotePythonValue(funneled) + } + if let resize = resize { + __kwargs["resize"] = RemotePythonValue(resize) + } + if let color = color { + __kwargs["color"] = RemotePythonValue(color) + } + if let slice_ = slice_ { + __kwargs["slice_"] = RemotePythonValue(slice_) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_lfw_pairs", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_lfw_pairs", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_covtype(data_home: String? = nil, download_if_missing: Bool? = nil, random_state: Int? = nil, shuffle: Bool? = nil, return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_covtype", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_covtype", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_rcv1(data_home: String? = nil, subset: String? = nil, download_if_missing: Bool? = nil, random_state: Int? = nil, shuffle: Bool? = nil, return_X_y: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_rcv1", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_rcv1", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_kddcup99(subset: String? = nil, data_home: String? = nil, shuffle: Bool? = nil, random_state: Int? = nil, percent10: Bool? = nil, download_if_missing: Bool? = nil, return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let subset = subset { + __kwargs["subset"] = RemotePythonValue(subset) + } + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let percent10 = percent10 { + __kwargs["percent10"] = RemotePythonValue(percent10) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_kddcup99", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_kddcup99", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_california_housing(data_home: String? = nil, download_if_missing: Bool? = nil, return_X_y: Bool? = nil, as_frame: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let download_if_missing = download_if_missing { + __kwargs["download_if_missing"] = RemotePythonValue(download_if_missing) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_california_housing", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_california_housing", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fetch_openml(name: String? = nil, version: String? = nil, data_id: Int? = nil, data_home: String? = nil, target_column: String? = nil, cache: Bool? = nil, return_X_y: Bool? = nil, as_frame: String? = nil, n_retries: Int? = nil, delay: Double? = nil, parser: String? = nil, read_csv_kwargs: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let version = version { + __kwargs["version"] = RemotePythonValue(version) + } + if let data_id = data_id { + __kwargs["data_id"] = RemotePythonValue(data_id) + } + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let target_column = target_column { + __kwargs["target_column"] = RemotePythonValue(target_column) + } + if let cache = cache { + __kwargs["cache"] = RemotePythonValue(cache) + } + if let return_X_y = return_X_y { + __kwargs["return_X_y"] = RemotePythonValue(return_X_y) + } + if let as_frame = as_frame { + __kwargs["as_frame"] = RemotePythonValue(as_frame) + } + if let n_retries = n_retries { + __kwargs["n_retries"] = RemotePythonValue(n_retries) + } + if let delay = delay { + __kwargs["delay"] = RemotePythonValue(delay) + } + if let parser = parser { + __kwargs["parser"] = RemotePythonValue(parser) + } + if let read_csv_kwargs = read_csv_kwargs { + __kwargs["read_csv_kwargs"] = RemotePythonValue(read_csv_kwargs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_openml", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "fetch_openml", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_data_home(data_home: String? = nil, worker: Int? = nil) async throws -> String { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.datasets", function: "get_data_home", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.datasets", function: "get_data_home", args: __args, kwargs: __kwargs) + } + + public func clear_data_home(data_home: String? = nil, worker: Int? = nil) async throws -> Void { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let data_home = data_home { + __kwargs["data_home"] = RemotePythonValue(data_home) + } + if let worker { + try await pool.invokeVoid(module: "sklearn.datasets", function: "clear_data_home", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "sklearn.datasets", function: "clear_data_home", args: __args, kwargs: __kwargs) + } + + public func dump_svmlight_file(X: any PythonConvertible, y: any PythonConvertible, f: any PythonConvertible, zero_based: Bool? = nil, comment: String? = nil, query_id: (any PythonConvertible)? = nil, multilabel: Bool? = nil, worker: Int? = nil) async throws -> Void { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + __args.append(RemotePythonValue(f)) + var __kwargs: [String: RemotePythonValue] = [:] + if let zero_based = zero_based { + __kwargs["zero_based"] = RemotePythonValue(zero_based) + } + if let comment = comment { + __kwargs["comment"] = RemotePythonValue(comment) + } + if let query_id = query_id { + __kwargs["query_id"] = RemotePythonValue(query_id) + } + if let multilabel = multilabel { + __kwargs["multilabel"] = RemotePythonValue(multilabel) + } + if let worker { + try await pool.invokeVoid(module: "sklearn.datasets", function: "dump_svmlight_file", args: __args, kwargs: __kwargs, worker: worker) + return + } + try await pool.invokeVoid(module: "sklearn.datasets", function: "dump_svmlight_file", args: __args, kwargs: __kwargs) + } + + public func load_svmlight_file(f: any PythonConvertible, n_features: Int? = nil, dtype: (any PythonConvertible)? = nil, multilabel: Bool? = nil, zero_based: String? = nil, query_id: Bool? = nil, offset: Int? = nil, length: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(f)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let multilabel = multilabel { + __kwargs["multilabel"] = RemotePythonValue(multilabel) + } + if let zero_based = zero_based { + __kwargs["zero_based"] = RemotePythonValue(zero_based) + } + if let query_id = query_id { + __kwargs["query_id"] = RemotePythonValue(query_id) + } + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let length = length { + __kwargs["length"] = RemotePythonValue(length) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_svmlight_file", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_svmlight_file", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func load_svmlight_files(files: any PythonConvertible, n_features: Int? = nil, dtype: (any PythonConvertible)? = nil, multilabel: Bool? = nil, zero_based: String? = nil, query_id: Bool? = nil, offset: Int? = nil, length: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(files)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_features = n_features { + __kwargs["n_features"] = RemotePythonValue(n_features) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let multilabel = multilabel { + __kwargs["multilabel"] = RemotePythonValue(multilabel) + } + if let zero_based = zero_based { + __kwargs["zero_based"] = RemotePythonValue(zero_based) + } + if let query_id = query_id { + __kwargs["query_id"] = RemotePythonValue(query_id) + } + if let offset = offset { + __kwargs["offset"] = RemotePythonValue(offset) + } + if let length = length { + __kwargs["length"] = RemotePythonValue(length) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_svmlight_files", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.datasets", function: "load_svmlight_files", args: __args, kwargs: __kwargs) + } + return __owned + } +} + +// MARK: - Sklearn.Decomposition + +extension PoolSklearn { + /// Pool-bound `sklearn.decomposition` namespace. Calls run on this pool. + public var decomposition: PoolSklearnDecomposition { PoolSklearnDecomposition(pool: pool) } +} + +public struct PoolSklearnDecomposition: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func PCA(n_components: Int? = nil, copy: Bool? = nil, whiten: Bool? = nil, svd_solver: String? = nil, tol: Double? = nil, iterated_power: String? = nil, n_oversamples: Int? = nil, power_iteration_normalizer: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolPCA { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let whiten = whiten { + __kwargs["whiten"] = RemotePythonValue(whiten) + } + if let svd_solver = svd_solver { + __kwargs["svd_solver"] = RemotePythonValue(svd_solver) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let iterated_power = iterated_power { + __kwargs["iterated_power"] = RemotePythonValue(iterated_power) + } + if let n_oversamples = n_oversamples { + __kwargs["n_oversamples"] = RemotePythonValue(n_oversamples) + } + if let power_iteration_normalizer = power_iteration_normalizer { + __kwargs["power_iteration_normalizer"] = RemotePythonValue(power_iteration_normalizer) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "PCA", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "PCA", args: __args, kwargs: __kwargs) + } + return PoolPCA(pool: pool, owned: __owned) + } + + public func IncrementalPCA(n_components: Int? = nil, whiten: Bool? = nil, copy: Bool? = nil, batch_size: Int? = nil, worker: Int? = nil) async throws -> PoolIncrementalPCA { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let whiten = whiten { + __kwargs["whiten"] = RemotePythonValue(whiten) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let batch_size = batch_size { + __kwargs["batch_size"] = RemotePythonValue(batch_size) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "IncrementalPCA", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "IncrementalPCA", args: __args, kwargs: __kwargs) + } + return PoolIncrementalPCA(pool: pool, owned: __owned) + } + + public func KernelPCA(n_components: Int? = nil, kernel: String? = nil, gamma: Double? = nil, degree: Int? = nil, coef0: Double? = nil, kernel_params: (any PythonConvertible)? = nil, alpha: Double? = nil, fit_inverse_transform: Bool? = nil, eigen_solver: String? = nil, tol: Double? = nil, max_iter: Int? = nil, iterated_power: String? = nil, remove_zero_eig: Bool? = nil, random_state: Int? = nil, copy_X: Bool? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolKernelPCA { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let kernel = kernel { + __kwargs["kernel"] = RemotePythonValue(kernel) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let kernel_params = kernel_params { + __kwargs["kernel_params"] = RemotePythonValue(kernel_params) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_inverse_transform = fit_inverse_transform { + __kwargs["fit_inverse_transform"] = RemotePythonValue(fit_inverse_transform) + } + if let eigen_solver = eigen_solver { + __kwargs["eigen_solver"] = RemotePythonValue(eigen_solver) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let iterated_power = iterated_power { + __kwargs["iterated_power"] = RemotePythonValue(iterated_power) + } + if let remove_zero_eig = remove_zero_eig { + __kwargs["remove_zero_eig"] = RemotePythonValue(remove_zero_eig) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "KernelPCA", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "KernelPCA", args: __args, kwargs: __kwargs) + } + return PoolKernelPCA(pool: pool, owned: __owned) + } + + public func SparsePCA(n_components: Int? = nil, alpha: Double? = nil, ridge_alpha: Double? = nil, max_iter: Int? = nil, tol: Double? = nil, method: String? = nil, n_jobs: Int? = nil, verbose: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolSparsePCA { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let ridge_alpha = ridge_alpha { + __kwargs["ridge_alpha"] = RemotePythonValue(ridge_alpha) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "SparsePCA", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "SparsePCA", args: __args, kwargs: __kwargs) + } + return PoolSparsePCA(pool: pool, owned: __owned) + } + + public func TruncatedSVD(n_components: Int? = nil, algorithm: String? = nil, n_iter: Int? = nil, n_oversamples: Int? = nil, power_iteration_normalizer: String? = nil, random_state: Int? = nil, tol: Double? = nil, worker: Int? = nil) async throws -> PoolTruncatedSVD { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let n_iter = n_iter { + __kwargs["n_iter"] = RemotePythonValue(n_iter) + } + if let n_oversamples = n_oversamples { + __kwargs["n_oversamples"] = RemotePythonValue(n_oversamples) + } + if let power_iteration_normalizer = power_iteration_normalizer { + __kwargs["power_iteration_normalizer"] = RemotePythonValue(power_iteration_normalizer) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "TruncatedSVD", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "TruncatedSVD", args: __args, kwargs: __kwargs) + } + return PoolTruncatedSVD(pool: pool, owned: __owned) + } + + public func NMF(n_components: Int? = nil, `init`: String? = nil, solver: String? = nil, beta_loss: String? = nil, tol: Double? = nil, max_iter: Int? = nil, random_state: Int? = nil, alpha_W: Double? = nil, alpha_H: String? = nil, l1_ratio: Double? = nil, verbose: Int? = nil, shuffle: Bool? = nil, worker: Int? = nil) async throws -> PoolNMF { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let beta_loss = beta_loss { + __kwargs["beta_loss"] = RemotePythonValue(beta_loss) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let alpha_W = alpha_W { + __kwargs["alpha_W"] = RemotePythonValue(alpha_W) + } + if let alpha_H = alpha_H { + __kwargs["alpha_H"] = RemotePythonValue(alpha_H) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "NMF", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "NMF", args: __args, kwargs: __kwargs) + } + return PoolNMF(pool: pool, owned: __owned) + } + + public func MiniBatchNMF(n_components: Int? = nil, `init`: String? = nil, batch_size: Int? = nil, beta_loss: String? = nil, tol: Double? = nil, max_no_improvement: Int? = nil, max_iter: Int? = nil, alpha_W: Double? = nil, alpha_H: String? = nil, l1_ratio: Double? = nil, forget_factor: Double? = nil, fresh_restarts: Bool? = nil, fresh_restarts_max_iter: Int? = nil, transform_max_iter: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolMiniBatchNMF { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let batch_size = batch_size { + __kwargs["batch_size"] = RemotePythonValue(batch_size) + } + if let beta_loss = beta_loss { + __kwargs["beta_loss"] = RemotePythonValue(beta_loss) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let max_no_improvement = max_no_improvement { + __kwargs["max_no_improvement"] = RemotePythonValue(max_no_improvement) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let alpha_W = alpha_W { + __kwargs["alpha_W"] = RemotePythonValue(alpha_W) + } + if let alpha_H = alpha_H { + __kwargs["alpha_H"] = RemotePythonValue(alpha_H) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let forget_factor = forget_factor { + __kwargs["forget_factor"] = RemotePythonValue(forget_factor) + } + if let fresh_restarts = fresh_restarts { + __kwargs["fresh_restarts"] = RemotePythonValue(fresh_restarts) + } + if let fresh_restarts_max_iter = fresh_restarts_max_iter { + __kwargs["fresh_restarts_max_iter"] = RemotePythonValue(fresh_restarts_max_iter) + } + if let transform_max_iter = transform_max_iter { + __kwargs["transform_max_iter"] = RemotePythonValue(transform_max_iter) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "MiniBatchNMF", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "MiniBatchNMF", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchNMF(pool: pool, owned: __owned) + } + + public func FastICA(n_components: Int? = nil, algorithm: String? = nil, whiten: String? = nil, fun: String? = nil, fun_args: (any PythonConvertible)? = nil, max_iter: Int? = nil, tol: Double? = nil, w_init: (any PythonConvertible)? = nil, whiten_solver: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolFastICA { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let whiten = whiten { + __kwargs["whiten"] = RemotePythonValue(whiten) + } + if let fun = fun { + __kwargs["fun"] = RemotePythonValue(fun) + } + if let fun_args = fun_args { + __kwargs["fun_args"] = RemotePythonValue(fun_args) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let w_init = w_init { + __kwargs["w_init"] = RemotePythonValue(w_init) + } + if let whiten_solver = whiten_solver { + __kwargs["whiten_solver"] = RemotePythonValue(whiten_solver) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "FastICA", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "FastICA", args: __args, kwargs: __kwargs) + } + return PoolFastICA(pool: pool, owned: __owned) + } + + public func FactorAnalysis(n_components: Int? = nil, tol: Double? = nil, copy: Bool? = nil, max_iter: Int? = nil, noise_variance_init: (any PythonConvertible)? = nil, svd_method: String? = nil, iterated_power: Int? = nil, rotation: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolFactorAnalysis { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let noise_variance_init = noise_variance_init { + __kwargs["noise_variance_init"] = RemotePythonValue(noise_variance_init) + } + if let svd_method = svd_method { + __kwargs["svd_method"] = RemotePythonValue(svd_method) + } + if let iterated_power = iterated_power { + __kwargs["iterated_power"] = RemotePythonValue(iterated_power) + } + if let rotation = rotation { + __kwargs["rotation"] = RemotePythonValue(rotation) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "FactorAnalysis", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "FactorAnalysis", args: __args, kwargs: __kwargs) + } + return PoolFactorAnalysis(pool: pool, owned: __owned) + } + + public func LatentDirichletAllocation(n_components: Int? = nil, doc_topic_prior: Double? = nil, topic_word_prior: Double? = nil, learning_method: String? = nil, learning_decay: Double? = nil, learning_offset: Double? = nil, max_iter: Int? = nil, batch_size: Int? = nil, evaluate_every: Int? = nil, total_samples: Int? = nil, perp_tol: Double? = nil, mean_change_tol: Double? = nil, max_doc_update_iter: Int? = nil, n_jobs: Int? = nil, verbose: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolLatentDirichletAllocation { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let doc_topic_prior = doc_topic_prior { + __kwargs["doc_topic_prior"] = RemotePythonValue(doc_topic_prior) + } + if let topic_word_prior = topic_word_prior { + __kwargs["topic_word_prior"] = RemotePythonValue(topic_word_prior) + } + if let learning_method = learning_method { + __kwargs["learning_method"] = RemotePythonValue(learning_method) + } + if let learning_decay = learning_decay { + __kwargs["learning_decay"] = RemotePythonValue(learning_decay) + } + if let learning_offset = learning_offset { + __kwargs["learning_offset"] = RemotePythonValue(learning_offset) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let batch_size = batch_size { + __kwargs["batch_size"] = RemotePythonValue(batch_size) + } + if let evaluate_every = evaluate_every { + __kwargs["evaluate_every"] = RemotePythonValue(evaluate_every) + } + if let total_samples = total_samples { + __kwargs["total_samples"] = RemotePythonValue(total_samples) + } + if let perp_tol = perp_tol { + __kwargs["perp_tol"] = RemotePythonValue(perp_tol) + } + if let mean_change_tol = mean_change_tol { + __kwargs["mean_change_tol"] = RemotePythonValue(mean_change_tol) + } + if let max_doc_update_iter = max_doc_update_iter { + __kwargs["max_doc_update_iter"] = RemotePythonValue(max_doc_update_iter) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "LatentDirichletAllocation", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "LatentDirichletAllocation", args: __args, kwargs: __kwargs) + } + return PoolLatentDirichletAllocation(pool: pool, owned: __owned) + } + + public func DictionaryLearning(n_components: Int? = nil, alpha: Double? = nil, max_iter: Int? = nil, tol: Double? = nil, fit_algorithm: String? = nil, transform_algorithm: String? = nil, transform_n_nonzero_coefs: Int? = nil, transform_alpha: Double? = nil, n_jobs: Int? = nil, code_init: (any PythonConvertible)? = nil, dict_init: (any PythonConvertible)? = nil, verbose: Int? = nil, split_sign: Bool? = nil, random_state: Int? = nil, positive_code: Bool? = nil, positive_dict: Bool? = nil, transform_max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolDictionaryLearning { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let fit_algorithm = fit_algorithm { + __kwargs["fit_algorithm"] = RemotePythonValue(fit_algorithm) + } + if let transform_algorithm = transform_algorithm { + __kwargs["transform_algorithm"] = RemotePythonValue(transform_algorithm) + } + if let transform_n_nonzero_coefs = transform_n_nonzero_coefs { + __kwargs["transform_n_nonzero_coefs"] = RemotePythonValue(transform_n_nonzero_coefs) + } + if let transform_alpha = transform_alpha { + __kwargs["transform_alpha"] = RemotePythonValue(transform_alpha) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let code_init = code_init { + __kwargs["code_init"] = RemotePythonValue(code_init) + } + if let dict_init = dict_init { + __kwargs["dict_init"] = RemotePythonValue(dict_init) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let split_sign = split_sign { + __kwargs["split_sign"] = RemotePythonValue(split_sign) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let positive_code = positive_code { + __kwargs["positive_code"] = RemotePythonValue(positive_code) + } + if let positive_dict = positive_dict { + __kwargs["positive_dict"] = RemotePythonValue(positive_dict) + } + if let transform_max_iter = transform_max_iter { + __kwargs["transform_max_iter"] = RemotePythonValue(transform_max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "DictionaryLearning", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "DictionaryLearning", args: __args, kwargs: __kwargs) + } + return PoolDictionaryLearning(pool: pool, owned: __owned) + } + + public func MiniBatchDictionaryLearning(n_components: Int? = nil, alpha: Double? = nil, n_iter: String? = nil, max_iter: Int? = nil, fit_algorithm: String? = nil, n_jobs: Int? = nil, batch_size: Int? = nil, shuffle: Bool? = nil, dict_init: (any PythonConvertible)? = nil, transform_algorithm: String? = nil, transform_n_nonzero_coefs: Int? = nil, transform_alpha: Double? = nil, verbose: Int? = nil, split_sign: Bool? = nil, random_state: Int? = nil, positive_code: Bool? = nil, positive_dict: Bool? = nil, transform_max_iter: Int? = nil, callback: (any PythonConvertible)? = nil, tol: Double? = nil, max_no_improvement: Int? = nil, worker: Int? = nil) async throws -> PoolMiniBatchDictionaryLearning { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_components = n_components { + __kwargs["n_components"] = RemotePythonValue(n_components) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let n_iter = n_iter { + __kwargs["n_iter"] = RemotePythonValue(n_iter) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let fit_algorithm = fit_algorithm { + __kwargs["fit_algorithm"] = RemotePythonValue(fit_algorithm) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let batch_size = batch_size { + __kwargs["batch_size"] = RemotePythonValue(batch_size) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let dict_init = dict_init { + __kwargs["dict_init"] = RemotePythonValue(dict_init) + } + if let transform_algorithm = transform_algorithm { + __kwargs["transform_algorithm"] = RemotePythonValue(transform_algorithm) + } + if let transform_n_nonzero_coefs = transform_n_nonzero_coefs { + __kwargs["transform_n_nonzero_coefs"] = RemotePythonValue(transform_n_nonzero_coefs) + } + if let transform_alpha = transform_alpha { + __kwargs["transform_alpha"] = RemotePythonValue(transform_alpha) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let split_sign = split_sign { + __kwargs["split_sign"] = RemotePythonValue(split_sign) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let positive_code = positive_code { + __kwargs["positive_code"] = RemotePythonValue(positive_code) + } + if let positive_dict = positive_dict { + __kwargs["positive_dict"] = RemotePythonValue(positive_dict) + } + if let transform_max_iter = transform_max_iter { + __kwargs["transform_max_iter"] = RemotePythonValue(transform_max_iter) + } + if let callback = callback { + __kwargs["callback"] = RemotePythonValue(callback) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let max_no_improvement = max_no_improvement { + __kwargs["max_no_improvement"] = RemotePythonValue(max_no_improvement) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "MiniBatchDictionaryLearning", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.decomposition", function: "MiniBatchDictionaryLearning", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchDictionaryLearning(pool: pool, owned: __owned) + } +} + + +public struct PoolPCA: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolPCA { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPCA(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_covariance(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_covariance", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_covariance", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_precision(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_precision", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_precision", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolIncrementalPCA: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolIncrementalPCA { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolIncrementalPCA(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolIncrementalPCA { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolIncrementalPCA(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_covariance(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_covariance", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_covariance", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_precision(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_precision", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_precision", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolKernelPCA: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolKernelPCA { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolKernelPCA(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolSparsePCA: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSparsePCA { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSparsePCA(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolTruncatedSVD: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolTruncatedSVD { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolTruncatedSVD(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolNMF: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolNMF { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolNMF(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolMiniBatchNMF: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMiniBatchNMF { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchNMF(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMiniBatchNMF { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchNMF(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolFastICA: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolFastICA { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolFastICA(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolFactorAnalysis: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolFactorAnalysis { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolFactorAnalysis(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_covariance(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_covariance", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_covariance", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_precision(worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_precision", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_precision", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolLatentDirichletAllocation: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolLatentDirichletAllocation { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLatentDirichletAllocation(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolLatentDirichletAllocation { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolLatentDirichletAllocation(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func perplexity(X: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "perplexity", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "perplexity", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolDictionaryLearning: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolDictionaryLearning { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolDictionaryLearning(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolMiniBatchDictionaryLearning: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMiniBatchDictionaryLearning { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchDictionaryLearning(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMiniBatchDictionaryLearning { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolMiniBatchDictionaryLearning(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + +// MARK: - Sklearn.Ensemble + +extension PoolSklearn { + /// Pool-bound `sklearn.ensemble` namespace. Calls run on this pool. + public var ensemble: PoolSklearnEnsemble { PoolSklearnEnsemble(pool: pool) } +} + +public struct PoolSklearnEnsemble: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func RandomForestClassifier(n_estimators: Int? = nil, criterion: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: String? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, bootstrap: Bool? = nil, oob_score: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, warm_start: Bool? = nil, class_weight: (any PythonConvertible)? = nil, ccp_alpha: Double? = nil, max_samples: Double? = nil, monotonic_cst: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolRandomForestClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let oob_score = oob_score { + __kwargs["oob_score"] = RemotePythonValue(oob_score) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + if let monotonic_cst = monotonic_cst { + __kwargs["monotonic_cst"] = RemotePythonValue(monotonic_cst) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "RandomForestClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "RandomForestClassifier", args: __args, kwargs: __kwargs) + } + return PoolRandomForestClassifier(pool: pool, owned: __owned) + } + + public func RandomForestRegressor(n_estimators: Int? = nil, criterion: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: Double? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, bootstrap: Bool? = nil, oob_score: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, warm_start: Bool? = nil, ccp_alpha: Double? = nil, max_samples: Double? = nil, monotonic_cst: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolRandomForestRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let oob_score = oob_score { + __kwargs["oob_score"] = RemotePythonValue(oob_score) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + if let monotonic_cst = monotonic_cst { + __kwargs["monotonic_cst"] = RemotePythonValue(monotonic_cst) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "RandomForestRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "RandomForestRegressor", args: __args, kwargs: __kwargs) + } + return PoolRandomForestRegressor(pool: pool, owned: __owned) + } + + public func ExtraTreesClassifier(n_estimators: Int? = nil, criterion: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: String? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, bootstrap: Bool? = nil, oob_score: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, warm_start: Bool? = nil, class_weight: (any PythonConvertible)? = nil, ccp_alpha: Double? = nil, max_samples: Double? = nil, worker: Int? = nil) async throws -> PoolExtraTreesClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let oob_score = oob_score { + __kwargs["oob_score"] = RemotePythonValue(oob_score) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "ExtraTreesClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "ExtraTreesClassifier", args: __args, kwargs: __kwargs) + } + return PoolExtraTreesClassifier(pool: pool, owned: __owned) + } + + public func ExtraTreesRegressor(n_estimators: Int? = nil, criterion: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: Double? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, bootstrap: Bool? = nil, oob_score: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, warm_start: Bool? = nil, ccp_alpha: Double? = nil, max_samples: Double? = nil, worker: Int? = nil) async throws -> PoolExtraTreesRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let oob_score = oob_score { + __kwargs["oob_score"] = RemotePythonValue(oob_score) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "ExtraTreesRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "ExtraTreesRegressor", args: __args, kwargs: __kwargs) + } + return PoolExtraTreesRegressor(pool: pool, owned: __owned) + } + + public func GradientBoostingClassifier(loss: String? = nil, learning_rate: Double? = nil, n_estimators: Int? = nil, subsample: Double? = nil, criterion: String? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_depth: Int? = nil, min_impurity_decrease: Double? = nil, `init`: (any PythonConvertible)? = nil, random_state: Int? = nil, max_features: (any PythonConvertible)? = nil, verbose: Int? = nil, max_leaf_nodes: Int? = nil, warm_start: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, tol: Double? = nil, ccp_alpha: Double? = nil, worker: Int? = nil) async throws -> PoolGradientBoostingClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let subsample = subsample { + __kwargs["subsample"] = RemotePythonValue(subsample) + } + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "GradientBoostingClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "GradientBoostingClassifier", args: __args, kwargs: __kwargs) + } + return PoolGradientBoostingClassifier(pool: pool, owned: __owned) + } + + public func GradientBoostingRegressor(loss: String? = nil, learning_rate: Double? = nil, n_estimators: Int? = nil, subsample: Double? = nil, criterion: String? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_depth: Int? = nil, min_impurity_decrease: Double? = nil, `init`: (any PythonConvertible)? = nil, random_state: Int? = nil, max_features: (any PythonConvertible)? = nil, alpha: Double? = nil, verbose: Int? = nil, max_leaf_nodes: Int? = nil, warm_start: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, tol: Double? = nil, ccp_alpha: Double? = nil, worker: Int? = nil) async throws -> PoolGradientBoostingRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let subsample = subsample { + __kwargs["subsample"] = RemotePythonValue(subsample) + } + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let `init` = `init` { + __kwargs["init"] = RemotePythonValue(`init`) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "GradientBoostingRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "GradientBoostingRegressor", args: __args, kwargs: __kwargs) + } + return PoolGradientBoostingRegressor(pool: pool, owned: __owned) + } + + public func HistGradientBoostingClassifier(loss: String? = nil, learning_rate: Double? = nil, max_iter: Int? = nil, max_leaf_nodes: Int? = nil, max_depth: Int? = nil, min_samples_leaf: Int? = nil, l2_regularization: Double? = nil, max_features: Double? = nil, max_bins: Int? = nil, categorical_features: (any PythonConvertible)? = nil, monotonic_cst: (any PythonConvertible)? = nil, interaction_cst: (any PythonConvertible)? = nil, warm_start: Bool? = nil, early_stopping: String? = nil, scoring: String? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, tol: Double? = nil, verbose: Int? = nil, random_state: Int? = nil, class_weight: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolHistGradientBoostingClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let l2_regularization = l2_regularization { + __kwargs["l2_regularization"] = RemotePythonValue(l2_regularization) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let max_bins = max_bins { + __kwargs["max_bins"] = RemotePythonValue(max_bins) + } + if let categorical_features = categorical_features { + __kwargs["categorical_features"] = RemotePythonValue(categorical_features) + } + if let monotonic_cst = monotonic_cst { + __kwargs["monotonic_cst"] = RemotePythonValue(monotonic_cst) + } + if let interaction_cst = interaction_cst { + __kwargs["interaction_cst"] = RemotePythonValue(interaction_cst) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "HistGradientBoostingClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "HistGradientBoostingClassifier", args: __args, kwargs: __kwargs) + } + return PoolHistGradientBoostingClassifier(pool: pool, owned: __owned) + } + + public func HistGradientBoostingRegressor(loss: String? = nil, quantile: Double? = nil, learning_rate: Double? = nil, max_iter: Int? = nil, max_leaf_nodes: Int? = nil, max_depth: Int? = nil, min_samples_leaf: Int? = nil, l2_regularization: Double? = nil, max_features: Double? = nil, max_bins: Int? = nil, categorical_features: (any PythonConvertible)? = nil, monotonic_cst: (any PythonConvertible)? = nil, interaction_cst: (any PythonConvertible)? = nil, warm_start: Bool? = nil, early_stopping: String? = nil, scoring: String? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, tol: Double? = nil, verbose: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolHistGradientBoostingRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let quantile = quantile { + __kwargs["quantile"] = RemotePythonValue(quantile) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let l2_regularization = l2_regularization { + __kwargs["l2_regularization"] = RemotePythonValue(l2_regularization) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let max_bins = max_bins { + __kwargs["max_bins"] = RemotePythonValue(max_bins) + } + if let categorical_features = categorical_features { + __kwargs["categorical_features"] = RemotePythonValue(categorical_features) + } + if let monotonic_cst = monotonic_cst { + __kwargs["monotonic_cst"] = RemotePythonValue(monotonic_cst) + } + if let interaction_cst = interaction_cst { + __kwargs["interaction_cst"] = RemotePythonValue(interaction_cst) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "HistGradientBoostingRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "HistGradientBoostingRegressor", args: __args, kwargs: __kwargs) + } + return PoolHistGradientBoostingRegressor(pool: pool, owned: __owned) + } + + public func AdaBoostClassifier(estimator: (any PythonConvertible)? = nil, n_estimators: Int? = nil, learning_rate: Double? = nil, algorithm: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolAdaBoostClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator = estimator { + __kwargs["estimator"] = RemotePythonValue(estimator) + } + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "AdaBoostClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "AdaBoostClassifier", args: __args, kwargs: __kwargs) + } + return PoolAdaBoostClassifier(pool: pool, owned: __owned) + } + + public func AdaBoostRegressor(estimator: (any PythonConvertible)? = nil, n_estimators: Int? = nil, learning_rate: Double? = nil, loss: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolAdaBoostRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator = estimator { + __kwargs["estimator"] = RemotePythonValue(estimator) + } + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "AdaBoostRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "AdaBoostRegressor", args: __args, kwargs: __kwargs) + } + return PoolAdaBoostRegressor(pool: pool, owned: __owned) + } + + public func BaggingClassifier(estimator: (any PythonConvertible)? = nil, n_estimators: Int? = nil, max_samples: Double? = nil, max_features: Double? = nil, bootstrap: Bool? = nil, bootstrap_features: Bool? = nil, oob_score: Bool? = nil, warm_start: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolBaggingClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator = estimator { + __kwargs["estimator"] = RemotePythonValue(estimator) + } + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let bootstrap_features = bootstrap_features { + __kwargs["bootstrap_features"] = RemotePythonValue(bootstrap_features) + } + if let oob_score = oob_score { + __kwargs["oob_score"] = RemotePythonValue(oob_score) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "BaggingClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "BaggingClassifier", args: __args, kwargs: __kwargs) + } + return PoolBaggingClassifier(pool: pool, owned: __owned) + } + + public func BaggingRegressor(estimator: (any PythonConvertible)? = nil, n_estimators: Int? = nil, max_samples: Double? = nil, max_features: Double? = nil, bootstrap: Bool? = nil, bootstrap_features: Bool? = nil, oob_score: Bool? = nil, warm_start: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolBaggingRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator = estimator { + __kwargs["estimator"] = RemotePythonValue(estimator) + } + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let bootstrap_features = bootstrap_features { + __kwargs["bootstrap_features"] = RemotePythonValue(bootstrap_features) + } + if let oob_score = oob_score { + __kwargs["oob_score"] = RemotePythonValue(oob_score) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "BaggingRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "BaggingRegressor", args: __args, kwargs: __kwargs) + } + return PoolBaggingRegressor(pool: pool, owned: __owned) + } + + public func VotingClassifier(estimators: PyHandle, voting: String? = nil, weights: [Double]? = nil, n_jobs: Int? = nil, flatten_transform: Bool? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolVotingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimators)) + var __kwargs: [String: RemotePythonValue] = [:] + if let voting = voting { + __kwargs["voting"] = RemotePythonValue(voting) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let flatten_transform = flatten_transform { + __kwargs["flatten_transform"] = RemotePythonValue(flatten_transform) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "VotingClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "VotingClassifier", args: __args, kwargs: __kwargs) + } + return PoolVotingClassifier(pool: pool, owned: __owned) + } + + public func VotingRegressor(estimators: PyHandle, weights: [Double]? = nil, n_jobs: Int? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolVotingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimators)) + var __kwargs: [String: RemotePythonValue] = [:] + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "VotingRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "VotingRegressor", args: __args, kwargs: __kwargs) + } + return PoolVotingRegressor(pool: pool, owned: __owned) + } + + public func StackingClassifier(estimators: PyHandle, final_estimator: (any PythonConvertible)? = nil, cv: Int? = nil, stack_method: String? = nil, n_jobs: Int? = nil, passthrough: Bool? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolStackingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimators)) + var __kwargs: [String: RemotePythonValue] = [:] + if let final_estimator = final_estimator { + __kwargs["final_estimator"] = RemotePythonValue(final_estimator) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let stack_method = stack_method { + __kwargs["stack_method"] = RemotePythonValue(stack_method) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let passthrough = passthrough { + __kwargs["passthrough"] = RemotePythonValue(passthrough) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "StackingClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "StackingClassifier", args: __args, kwargs: __kwargs) + } + return PoolStackingClassifier(pool: pool, owned: __owned) + } + + public func StackingRegressor(estimators: PyHandle, final_estimator: (any PythonConvertible)? = nil, cv: Int? = nil, n_jobs: Int? = nil, passthrough: Bool? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolStackingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimators)) + var __kwargs: [String: RemotePythonValue] = [:] + if let final_estimator = final_estimator { + __kwargs["final_estimator"] = RemotePythonValue(final_estimator) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let passthrough = passthrough { + __kwargs["passthrough"] = RemotePythonValue(passthrough) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "StackingRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "StackingRegressor", args: __args, kwargs: __kwargs) + } + return PoolStackingRegressor(pool: pool, owned: __owned) + } + + public func IsolationForest(n_estimators: Int? = nil, max_samples: String? = nil, contamination: String? = nil, max_features: Double? = nil, bootstrap: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, warm_start: Bool? = nil, worker: Int? = nil) async throws -> PoolIsolationForest { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let max_samples = max_samples { + __kwargs["max_samples"] = RemotePythonValue(max_samples) + } + if let contamination = contamination { + __kwargs["contamination"] = RemotePythonValue(contamination) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let bootstrap = bootstrap { + __kwargs["bootstrap"] = RemotePythonValue(bootstrap) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "IsolationForest", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "IsolationForest", args: __args, kwargs: __kwargs) + } + return PoolIsolationForest(pool: pool, owned: __owned) + } + + public func RandomTreesEmbedding(n_estimators: Int? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, sparse_output: Bool? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, warm_start: Bool? = nil, worker: Int? = nil) async throws -> PoolRandomTreesEmbedding { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_estimators = n_estimators { + __kwargs["n_estimators"] = RemotePythonValue(n_estimators) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let sparse_output = sparse_output { + __kwargs["sparse_output"] = RemotePythonValue(sparse_output) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "RandomTreesEmbedding", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.ensemble", function: "RandomTreesEmbedding", args: __args, kwargs: __kwargs) + } + return PoolRandomTreesEmbedding(pool: pool, owned: __owned) + } +} + + +public struct PoolRandomForestClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolRandomForestClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRandomForestClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolRandomForestRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolRandomForestRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRandomForestRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolExtraTreesClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolExtraTreesClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolExtraTreesClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolExtraTreesRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolExtraTreesRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolExtraTreesRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolGradientBoostingClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolGradientBoostingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolGradientBoostingClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolGradientBoostingRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolGradientBoostingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolGradientBoostingRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolHistGradientBoostingClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolHistGradientBoostingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolHistGradientBoostingClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolHistGradientBoostingRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolHistGradientBoostingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolHistGradientBoostingRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolAdaBoostClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolAdaBoostClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolAdaBoostClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func staged_score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_score", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_score", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolAdaBoostRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolAdaBoostRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolAdaBoostRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func staged_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func staged_score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "staged_score", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "staged_score", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolBaggingClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolBaggingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBaggingClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolBaggingRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolBaggingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBaggingRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolVotingClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolVotingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolVotingClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolVotingRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolVotingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolVotingRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolStackingClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolStackingClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolStackingClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolStackingRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolStackingRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolStackingRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolIsolationForest: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolIsolationForest { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolIsolationForest(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolRandomTreesEmbedding: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolRandomTreesEmbedding { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRandomTreesEmbedding(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } +} + +// MARK: - Sklearn.LinearModel + +extension PoolSklearn { + /// Pool-bound `sklearn.linear_model` namespace. Calls run on this pool. + public var linearmodel: PoolSklearnLinearmodel { PoolSklearnLinearmodel(pool: pool) } +} + +public struct PoolSklearnLinearmodel: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func lasso_path(X: any PythonConvertible, y: [Double], eps: Double? = nil, n_alphas: Int? = nil, alphas: (any PythonConvertible)? = nil, precompute: String? = nil, copy_X: Bool? = nil, coef_init: (any PythonConvertible)? = nil, verbose: Bool? = nil, return_n_iter: Bool? = nil, positive: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let n_alphas = n_alphas { + __kwargs["n_alphas"] = RemotePythonValue(n_alphas) + } + if let alphas = alphas { + __kwargs["alphas"] = RemotePythonValue(alphas) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let coef_init = coef_init { + __kwargs["coef_init"] = RemotePythonValue(coef_init) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "lasso_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "lasso_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func lars_path(X: any PythonConvertible, y: [Double], Xy: (any PythonConvertible)? = nil, Gram: (any PythonConvertible)? = nil, max_iter: Int? = nil, alpha_min: Double? = nil, method: String? = nil, copy_X: Bool? = nil, eps: Double? = nil, copy_Gram: Bool? = nil, verbose: Int? = nil, return_path: Bool? = nil, return_n_iter: Bool? = nil, positive: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + if let Xy = Xy { + __args.append(RemotePythonValue(Xy)) + } + if let Gram = Gram { + __args.append(RemotePythonValue(Gram)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let alpha_min = alpha_min { + __kwargs["alpha_min"] = RemotePythonValue(alpha_min) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let copy_Gram = copy_Gram { + __kwargs["copy_Gram"] = RemotePythonValue(copy_Gram) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let return_path = return_path { + __kwargs["return_path"] = RemotePythonValue(return_path) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "lars_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "lars_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func enet_path(X: any PythonConvertible, y: [Double], l1_ratio: Double? = nil, eps: Double? = nil, n_alphas: Int? = nil, alphas: (any PythonConvertible)? = nil, precompute: String? = nil, copy_X: Bool? = nil, coef_init: (any PythonConvertible)? = nil, verbose: Bool? = nil, return_n_iter: Bool? = nil, positive: Bool? = nil, check_input: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let n_alphas = n_alphas { + __kwargs["n_alphas"] = RemotePythonValue(n_alphas) + } + if let alphas = alphas { + __kwargs["alphas"] = RemotePythonValue(alphas) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let coef_init = coef_init { + __kwargs["coef_init"] = RemotePythonValue(coef_init) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let check_input = check_input { + __kwargs["check_input"] = RemotePythonValue(check_input) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "enet_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "enet_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func orthogonal_mp(X: any PythonConvertible, y: [Double], n_nonzero_coefs: Int? = nil, tol: Double? = nil, precompute: Bool? = nil, copy_X: Bool? = nil, return_path: Bool? = nil, return_n_iter: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_nonzero_coefs = n_nonzero_coefs { + __kwargs["n_nonzero_coefs"] = RemotePythonValue(n_nonzero_coefs) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let return_path = return_path { + __kwargs["return_path"] = RemotePythonValue(return_path) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "orthogonal_mp", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "orthogonal_mp", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func orthogonal_mp_gram(Gram: any PythonConvertible, Xy: any PythonConvertible, n_nonzero_coefs: Int? = nil, tol: Double? = nil, norms_squared: (any PythonConvertible)? = nil, copy_Gram: Bool? = nil, copy_Xy: Bool? = nil, return_path: Bool? = nil, return_n_iter: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(Gram)) + __args.append(RemotePythonValue(Xy)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_nonzero_coefs = n_nonzero_coefs { + __kwargs["n_nonzero_coefs"] = RemotePythonValue(n_nonzero_coefs) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let norms_squared = norms_squared { + __kwargs["norms_squared"] = RemotePythonValue(norms_squared) + } + if let copy_Gram = copy_Gram { + __kwargs["copy_Gram"] = RemotePythonValue(copy_Gram) + } + if let copy_Xy = copy_Xy { + __kwargs["copy_Xy"] = RemotePythonValue(copy_Xy) + } + if let return_path = return_path { + __kwargs["return_path"] = RemotePythonValue(return_path) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "orthogonal_mp_gram", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "orthogonal_mp_gram", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func ridge_regression(X: any PythonConvertible, y: [Double], alpha: Double, sample_weight: (any PythonConvertible)? = nil, solver: String? = nil, max_iter: Int? = nil, tol: Double? = nil, verbose: Int? = nil, positive: Bool? = nil, random_state: Int? = nil, return_n_iter: Bool? = nil, return_intercept: Bool? = nil, check_input: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + __args.append(RemotePythonValue(alpha)) + if let sample_weight = sample_weight { + __args.append(RemotePythonValue(sample_weight)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let return_n_iter = return_n_iter { + __kwargs["return_n_iter"] = RemotePythonValue(return_n_iter) + } + if let return_intercept = return_intercept { + __kwargs["return_intercept"] = RemotePythonValue(return_intercept) + } + if let check_input = check_input { + __kwargs["check_input"] = RemotePythonValue(check_input) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ridge_regression", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ridge_regression", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func LinearRegression(fit_intercept: Bool? = nil, copy_X: Bool? = nil, n_jobs: Int? = nil, positive: Bool? = nil, worker: Int? = nil) async throws -> PoolLinearRegression { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LinearRegression", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LinearRegression", args: __args, kwargs: __kwargs) + } + return PoolLinearRegression(pool: pool, owned: __owned) + } + + public func Ridge(alpha: Double? = nil, fit_intercept: Bool? = nil, copy_X: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, solver: String? = nil, positive: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolRidge { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Ridge", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Ridge", args: __args, kwargs: __kwargs) + } + return PoolRidge(pool: pool, owned: __owned) + } + + public func RidgeCV(alphas: (any PythonConvertible)? = nil, fit_intercept: Bool? = nil, scoring: String? = nil, cv: Int? = nil, gcv_mode: String? = nil, store_cv_values: Bool? = nil, alpha_per_target: Bool? = nil, worker: Int? = nil) async throws -> PoolRidgeCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alphas = alphas { + __kwargs["alphas"] = RemotePythonValue(alphas) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let gcv_mode = gcv_mode { + __kwargs["gcv_mode"] = RemotePythonValue(gcv_mode) + } + if let store_cv_values = store_cv_values { + __kwargs["store_cv_values"] = RemotePythonValue(store_cv_values) + } + if let alpha_per_target = alpha_per_target { + __kwargs["alpha_per_target"] = RemotePythonValue(alpha_per_target) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RidgeCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RidgeCV", args: __args, kwargs: __kwargs) + } + return PoolRidgeCV(pool: pool, owned: __owned) + } + + public func RidgeClassifier(alpha: Double? = nil, fit_intercept: Bool? = nil, copy_X: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, class_weight: (any PythonConvertible)? = nil, solver: String? = nil, positive: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolRidgeClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RidgeClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RidgeClassifier", args: __args, kwargs: __kwargs) + } + return PoolRidgeClassifier(pool: pool, owned: __owned) + } + + public func RidgeClassifierCV(alphas: (any PythonConvertible)? = nil, fit_intercept: Bool? = nil, scoring: String? = nil, cv: Int? = nil, class_weight: (any PythonConvertible)? = nil, store_cv_values: Bool? = nil, worker: Int? = nil) async throws -> PoolRidgeClassifierCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alphas = alphas { + __kwargs["alphas"] = RemotePythonValue(alphas) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let store_cv_values = store_cv_values { + __kwargs["store_cv_values"] = RemotePythonValue(store_cv_values) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RidgeClassifierCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RidgeClassifierCV", args: __args, kwargs: __kwargs) + } + return PoolRidgeClassifierCV(pool: pool, owned: __owned) + } + + public func Lasso(alpha: Double? = nil, fit_intercept: Bool? = nil, precompute: Bool? = nil, copy_X: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, warm_start: Bool? = nil, positive: Bool? = nil, random_state: Int? = nil, selection: String? = nil, worker: Int? = nil) async throws -> PoolLasso { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let selection = selection { + __kwargs["selection"] = RemotePythonValue(selection) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Lasso", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Lasso", args: __args, kwargs: __kwargs) + } + return PoolLasso(pool: pool, owned: __owned) + } + + public func LassoCV(eps: Double? = nil, n_alphas: Int? = nil, alphas: (any PythonConvertible)? = nil, fit_intercept: Bool? = nil, precompute: String? = nil, max_iter: Int? = nil, tol: Double? = nil, copy_X: Bool? = nil, cv: Int? = nil, verbose: Bool? = nil, n_jobs: Int? = nil, positive: Bool? = nil, random_state: Int? = nil, selection: String? = nil, worker: Int? = nil) async throws -> PoolLassoCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let n_alphas = n_alphas { + __kwargs["n_alphas"] = RemotePythonValue(n_alphas) + } + if let alphas = alphas { + __kwargs["alphas"] = RemotePythonValue(alphas) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let selection = selection { + __kwargs["selection"] = RemotePythonValue(selection) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoCV", args: __args, kwargs: __kwargs) + } + return PoolLassoCV(pool: pool, owned: __owned) + } + + public func LassoLars(alpha: Double? = nil, fit_intercept: Bool? = nil, verbose: Bool? = nil, precompute: String? = nil, max_iter: Int? = nil, eps: Double? = nil, copy_X: Bool? = nil, fit_path: Bool? = nil, positive: Bool? = nil, jitter: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolLassoLars { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let fit_path = fit_path { + __kwargs["fit_path"] = RemotePythonValue(fit_path) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let jitter = jitter { + __kwargs["jitter"] = RemotePythonValue(jitter) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoLars", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoLars", args: __args, kwargs: __kwargs) + } + return PoolLassoLars(pool: pool, owned: __owned) + } + + public func LassoLarsCV(fit_intercept: Bool? = nil, verbose: Bool? = nil, max_iter: Int? = nil, precompute: String? = nil, cv: Int? = nil, max_n_alphas: Int? = nil, n_jobs: Int? = nil, eps: Double? = nil, copy_X: Bool? = nil, positive: Bool? = nil, worker: Int? = nil) async throws -> PoolLassoLarsCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let max_n_alphas = max_n_alphas { + __kwargs["max_n_alphas"] = RemotePythonValue(max_n_alphas) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoLarsCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoLarsCV", args: __args, kwargs: __kwargs) + } + return PoolLassoLarsCV(pool: pool, owned: __owned) + } + + public func LassoLarsIC(criterion: String? = nil, fit_intercept: Bool? = nil, verbose: Bool? = nil, precompute: String? = nil, max_iter: Int? = nil, eps: Double? = nil, copy_X: Bool? = nil, positive: Bool? = nil, noise_variance: Double? = nil, worker: Int? = nil) async throws -> PoolLassoLarsIC { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let noise_variance = noise_variance { + __kwargs["noise_variance"] = RemotePythonValue(noise_variance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoLarsIC", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LassoLarsIC", args: __args, kwargs: __kwargs) + } + return PoolLassoLarsIC(pool: pool, owned: __owned) + } + + public func ElasticNet(alpha: Double? = nil, l1_ratio: Double? = nil, fit_intercept: Bool? = nil, precompute: Bool? = nil, max_iter: Int? = nil, copy_X: Bool? = nil, tol: Double? = nil, warm_start: Bool? = nil, positive: Bool? = nil, random_state: Int? = nil, selection: String? = nil, worker: Int? = nil) async throws -> PoolElasticNet { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let selection = selection { + __kwargs["selection"] = RemotePythonValue(selection) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ElasticNet", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ElasticNet", args: __args, kwargs: __kwargs) + } + return PoolElasticNet(pool: pool, owned: __owned) + } + + public func ElasticNetCV(l1_ratio: Double? = nil, eps: Double? = nil, n_alphas: Int? = nil, alphas: (any PythonConvertible)? = nil, fit_intercept: Bool? = nil, precompute: String? = nil, max_iter: Int? = nil, tol: Double? = nil, cv: Int? = nil, copy_X: Bool? = nil, verbose: Int? = nil, n_jobs: Int? = nil, positive: Bool? = nil, random_state: Int? = nil, selection: String? = nil, worker: Int? = nil) async throws -> PoolElasticNetCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let n_alphas = n_alphas { + __kwargs["n_alphas"] = RemotePythonValue(n_alphas) + } + if let alphas = alphas { + __kwargs["alphas"] = RemotePythonValue(alphas) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let positive = positive { + __kwargs["positive"] = RemotePythonValue(positive) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let selection = selection { + __kwargs["selection"] = RemotePythonValue(selection) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ElasticNetCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ElasticNetCV", args: __args, kwargs: __kwargs) + } + return PoolElasticNetCV(pool: pool, owned: __owned) + } + + public func MultiTaskLasso(alpha: Double? = nil, fit_intercept: Bool? = nil, copy_X: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, warm_start: Bool? = nil, random_state: Int? = nil, selection: String? = nil, worker: Int? = nil) async throws -> PoolMultiTaskLasso { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let selection = selection { + __kwargs["selection"] = RemotePythonValue(selection) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "MultiTaskLasso", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "MultiTaskLasso", args: __args, kwargs: __kwargs) + } + return PoolMultiTaskLasso(pool: pool, owned: __owned) + } + + public func MultiTaskElasticNet(alpha: Double? = nil, l1_ratio: Double? = nil, fit_intercept: Bool? = nil, copy_X: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, warm_start: Bool? = nil, random_state: Int? = nil, selection: String? = nil, worker: Int? = nil) async throws -> PoolMultiTaskElasticNet { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let selection = selection { + __kwargs["selection"] = RemotePythonValue(selection) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "MultiTaskElasticNet", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "MultiTaskElasticNet", args: __args, kwargs: __kwargs) + } + return PoolMultiTaskElasticNet(pool: pool, owned: __owned) + } + + public func Lars(fit_intercept: Bool? = nil, verbose: Bool? = nil, precompute: String? = nil, n_nonzero_coefs: Int? = nil, eps: Double? = nil, copy_X: Bool? = nil, fit_path: Bool? = nil, jitter: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolLars { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let n_nonzero_coefs = n_nonzero_coefs { + __kwargs["n_nonzero_coefs"] = RemotePythonValue(n_nonzero_coefs) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let fit_path = fit_path { + __kwargs["fit_path"] = RemotePythonValue(fit_path) + } + if let jitter = jitter { + __kwargs["jitter"] = RemotePythonValue(jitter) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Lars", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Lars", args: __args, kwargs: __kwargs) + } + return PoolLars(pool: pool, owned: __owned) + } + + public func LarsCV(fit_intercept: Bool? = nil, verbose: Bool? = nil, max_iter: Int? = nil, precompute: String? = nil, cv: Int? = nil, max_n_alphas: Int? = nil, n_jobs: Int? = nil, eps: Double? = nil, copy_X: Bool? = nil, worker: Int? = nil) async throws -> PoolLarsCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let max_n_alphas = max_n_alphas { + __kwargs["max_n_alphas"] = RemotePythonValue(max_n_alphas) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LarsCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LarsCV", args: __args, kwargs: __kwargs) + } + return PoolLarsCV(pool: pool, owned: __owned) + } + + public func LogisticRegression(penalty: String? = nil, dual: Bool? = nil, tol: Double? = nil, C: Double? = nil, fit_intercept: Bool? = nil, intercept_scaling: Double? = nil, class_weight: (any PythonConvertible)? = nil, random_state: Int? = nil, solver: String? = nil, max_iter: Int? = nil, multi_class: String? = nil, verbose: Int? = nil, warm_start: Bool? = nil, n_jobs: Int? = nil, l1_ratio: Double? = nil, worker: Int? = nil) async throws -> PoolLogisticRegression { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let penalty = penalty { + __kwargs["penalty"] = RemotePythonValue(penalty) + } + if let dual = dual { + __kwargs["dual"] = RemotePythonValue(dual) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let intercept_scaling = intercept_scaling { + __kwargs["intercept_scaling"] = RemotePythonValue(intercept_scaling) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let multi_class = multi_class { + __kwargs["multi_class"] = RemotePythonValue(multi_class) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LogisticRegression", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LogisticRegression", args: __args, kwargs: __kwargs) + } + return PoolLogisticRegression(pool: pool, owned: __owned) + } + + public func LogisticRegressionCV(Cs: Int? = nil, fit_intercept: Bool? = nil, cv: Int? = nil, dual: Bool? = nil, penalty: String? = nil, scoring: String? = nil, solver: String? = nil, tol: Double? = nil, max_iter: Int? = nil, class_weight: (any PythonConvertible)? = nil, n_jobs: Int? = nil, verbose: Int? = nil, refit: Bool? = nil, intercept_scaling: Double? = nil, multi_class: String? = nil, random_state: Int? = nil, l1_ratios: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolLogisticRegressionCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let Cs = Cs { + __kwargs["Cs"] = RemotePythonValue(Cs) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let dual = dual { + __kwargs["dual"] = RemotePythonValue(dual) + } + if let penalty = penalty { + __kwargs["penalty"] = RemotePythonValue(penalty) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let refit = refit { + __kwargs["refit"] = RemotePythonValue(refit) + } + if let intercept_scaling = intercept_scaling { + __kwargs["intercept_scaling"] = RemotePythonValue(intercept_scaling) + } + if let multi_class = multi_class { + __kwargs["multi_class"] = RemotePythonValue(multi_class) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let l1_ratios = l1_ratios { + __kwargs["l1_ratios"] = RemotePythonValue(l1_ratios) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LogisticRegressionCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "LogisticRegressionCV", args: __args, kwargs: __kwargs) + } + return PoolLogisticRegressionCV(pool: pool, owned: __owned) + } + + public func SGDClassifier(loss: String? = nil, penalty: String? = nil, alpha: Double? = nil, l1_ratio: Double? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, shuffle: Bool? = nil, verbose: Int? = nil, epsilon: Double? = nil, n_jobs: Int? = nil, random_state: Int? = nil, learning_rate: String? = nil, eta0: Double? = nil, power_t: Double? = nil, early_stopping: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, class_weight: (any PythonConvertible)? = nil, warm_start: Bool? = nil, average: Bool? = nil, worker: Int? = nil) async throws -> PoolSGDClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let penalty = penalty { + __kwargs["penalty"] = RemotePythonValue(penalty) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let epsilon = epsilon { + __kwargs["epsilon"] = RemotePythonValue(epsilon) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let eta0 = eta0 { + __kwargs["eta0"] = RemotePythonValue(eta0) + } + if let power_t = power_t { + __kwargs["power_t"] = RemotePythonValue(power_t) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "SGDClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "SGDClassifier", args: __args, kwargs: __kwargs) + } + return PoolSGDClassifier(pool: pool, owned: __owned) + } + + public func SGDRegressor(loss: String? = nil, penalty: String? = nil, alpha: Double? = nil, l1_ratio: Double? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, shuffle: Bool? = nil, verbose: Int? = nil, epsilon: Double? = nil, random_state: Int? = nil, learning_rate: String? = nil, eta0: Double? = nil, power_t: Double? = nil, early_stopping: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, warm_start: Bool? = nil, average: Bool? = nil, worker: Int? = nil) async throws -> PoolSGDRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let penalty = penalty { + __kwargs["penalty"] = RemotePythonValue(penalty) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let epsilon = epsilon { + __kwargs["epsilon"] = RemotePythonValue(epsilon) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let eta0 = eta0 { + __kwargs["eta0"] = RemotePythonValue(eta0) + } + if let power_t = power_t { + __kwargs["power_t"] = RemotePythonValue(power_t) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "SGDRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "SGDRegressor", args: __args, kwargs: __kwargs) + } + return PoolSGDRegressor(pool: pool, owned: __owned) + } + + public func SGDOneClassSVM(nu: Double? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, shuffle: Bool? = nil, verbose: Int? = nil, random_state: Int? = nil, learning_rate: String? = nil, eta0: Double? = nil, power_t: Double? = nil, warm_start: Bool? = nil, average: Bool? = nil, worker: Int? = nil) async throws -> PoolSGDOneClassSVM { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let nu = nu { + __kwargs["nu"] = RemotePythonValue(nu) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let learning_rate = learning_rate { + __kwargs["learning_rate"] = RemotePythonValue(learning_rate) + } + if let eta0 = eta0 { + __kwargs["eta0"] = RemotePythonValue(eta0) + } + if let power_t = power_t { + __kwargs["power_t"] = RemotePythonValue(power_t) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "SGDOneClassSVM", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "SGDOneClassSVM", args: __args, kwargs: __kwargs) + } + return PoolSGDOneClassSVM(pool: pool, owned: __owned) + } + + public func Perceptron(penalty: String? = nil, alpha: Double? = nil, l1_ratio: Double? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, shuffle: Bool? = nil, verbose: Int? = nil, eta0: Double? = nil, n_jobs: Int? = nil, random_state: Int? = nil, early_stopping: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, class_weight: (any PythonConvertible)? = nil, warm_start: Bool? = nil, worker: Int? = nil) async throws -> PoolPerceptron { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let penalty = penalty { + __kwargs["penalty"] = RemotePythonValue(penalty) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let l1_ratio = l1_ratio { + __kwargs["l1_ratio"] = RemotePythonValue(l1_ratio) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let eta0 = eta0 { + __kwargs["eta0"] = RemotePythonValue(eta0) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Perceptron", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "Perceptron", args: __args, kwargs: __kwargs) + } + return PoolPerceptron(pool: pool, owned: __owned) + } + + public func PassiveAggressiveClassifier(C: Double? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, early_stopping: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, shuffle: Bool? = nil, verbose: Int? = nil, loss: String? = nil, n_jobs: Int? = nil, random_state: Int? = nil, warm_start: Bool? = nil, class_weight: (any PythonConvertible)? = nil, average: Bool? = nil, worker: Int? = nil) async throws -> PoolPassiveAggressiveClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "PassiveAggressiveClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "PassiveAggressiveClassifier", args: __args, kwargs: __kwargs) + } + return PoolPassiveAggressiveClassifier(pool: pool, owned: __owned) + } + + public func PassiveAggressiveRegressor(C: Double? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, tol: Double? = nil, early_stopping: Bool? = nil, validation_fraction: Double? = nil, n_iter_no_change: Int? = nil, shuffle: Bool? = nil, verbose: Int? = nil, loss: String? = nil, epsilon: Double? = nil, random_state: Int? = nil, warm_start: Bool? = nil, average: Bool? = nil, worker: Int? = nil) async throws -> PoolPassiveAggressiveRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let early_stopping = early_stopping { + __kwargs["early_stopping"] = RemotePythonValue(early_stopping) + } + if let validation_fraction = validation_fraction { + __kwargs["validation_fraction"] = RemotePythonValue(validation_fraction) + } + if let n_iter_no_change = n_iter_no_change { + __kwargs["n_iter_no_change"] = RemotePythonValue(n_iter_no_change) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let epsilon = epsilon { + __kwargs["epsilon"] = RemotePythonValue(epsilon) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "PassiveAggressiveRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "PassiveAggressiveRegressor", args: __args, kwargs: __kwargs) + } + return PoolPassiveAggressiveRegressor(pool: pool, owned: __owned) + } + + public func BayesianRidge(max_iter: Int? = nil, tol: Double? = nil, alpha_1: Double? = nil, alpha_2: Double? = nil, lambda_1: Double? = nil, lambda_2: Double? = nil, alpha_init: Double? = nil, lambda_init: Double? = nil, compute_score: Bool? = nil, fit_intercept: Bool? = nil, copy_X: Bool? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolBayesianRidge { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let alpha_1 = alpha_1 { + __kwargs["alpha_1"] = RemotePythonValue(alpha_1) + } + if let alpha_2 = alpha_2 { + __kwargs["alpha_2"] = RemotePythonValue(alpha_2) + } + if let lambda_1 = lambda_1 { + __kwargs["lambda_1"] = RemotePythonValue(lambda_1) + } + if let lambda_2 = lambda_2 { + __kwargs["lambda_2"] = RemotePythonValue(lambda_2) + } + if let alpha_init = alpha_init { + __kwargs["alpha_init"] = RemotePythonValue(alpha_init) + } + if let lambda_init = lambda_init { + __kwargs["lambda_init"] = RemotePythonValue(lambda_init) + } + if let compute_score = compute_score { + __kwargs["compute_score"] = RemotePythonValue(compute_score) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "BayesianRidge", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "BayesianRidge", args: __args, kwargs: __kwargs) + } + return PoolBayesianRidge(pool: pool, owned: __owned) + } + + public func ARDRegression(max_iter: Int? = nil, tol: Double? = nil, alpha_1: Double? = nil, alpha_2: Double? = nil, lambda_1: Double? = nil, lambda_2: Double? = nil, compute_score: Bool? = nil, threshold_lambda: Double? = nil, fit_intercept: Bool? = nil, copy_X: Bool? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolARDRegression { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let alpha_1 = alpha_1 { + __kwargs["alpha_1"] = RemotePythonValue(alpha_1) + } + if let alpha_2 = alpha_2 { + __kwargs["alpha_2"] = RemotePythonValue(alpha_2) + } + if let lambda_1 = lambda_1 { + __kwargs["lambda_1"] = RemotePythonValue(lambda_1) + } + if let lambda_2 = lambda_2 { + __kwargs["lambda_2"] = RemotePythonValue(lambda_2) + } + if let compute_score = compute_score { + __kwargs["compute_score"] = RemotePythonValue(compute_score) + } + if let threshold_lambda = threshold_lambda { + __kwargs["threshold_lambda"] = RemotePythonValue(threshold_lambda) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ARDRegression", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "ARDRegression", args: __args, kwargs: __kwargs) + } + return PoolARDRegression(pool: pool, owned: __owned) + } + + public func HuberRegressor(epsilon: Double? = nil, max_iter: Int? = nil, alpha: Double? = nil, warm_start: Bool? = nil, fit_intercept: Bool? = nil, tol: Double? = nil, worker: Int? = nil) async throws -> PoolHuberRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let epsilon = epsilon { + __kwargs["epsilon"] = RemotePythonValue(epsilon) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "HuberRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "HuberRegressor", args: __args, kwargs: __kwargs) + } + return PoolHuberRegressor(pool: pool, owned: __owned) + } + + public func QuantileRegressor(quantile: Double? = nil, alpha: Double? = nil, fit_intercept: Bool? = nil, solver: String? = nil, solver_options: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolQuantileRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let quantile = quantile { + __kwargs["quantile"] = RemotePythonValue(quantile) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let solver_options = solver_options { + __kwargs["solver_options"] = RemotePythonValue(solver_options) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "QuantileRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "QuantileRegressor", args: __args, kwargs: __kwargs) + } + return PoolQuantileRegressor(pool: pool, owned: __owned) + } + + public func RANSACRegressor(estimator: (any PythonConvertible)? = nil, min_samples: Int? = nil, residual_threshold: Double? = nil, is_data_valid: (any PythonConvertible)? = nil, is_model_valid: (any PythonConvertible)? = nil, max_trials: Int? = nil, max_skips: Int? = nil, stop_n_inliers: Int? = nil, stop_score: Double? = nil, stop_probability: Double? = nil, loss: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolRANSACRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator = estimator { + __kwargs["estimator"] = RemotePythonValue(estimator) + } + if let min_samples = min_samples { + __kwargs["min_samples"] = RemotePythonValue(min_samples) + } + if let residual_threshold = residual_threshold { + __kwargs["residual_threshold"] = RemotePythonValue(residual_threshold) + } + if let is_data_valid = is_data_valid { + __kwargs["is_data_valid"] = RemotePythonValue(is_data_valid) + } + if let is_model_valid = is_model_valid { + __kwargs["is_model_valid"] = RemotePythonValue(is_model_valid) + } + if let max_trials = max_trials { + __kwargs["max_trials"] = RemotePythonValue(max_trials) + } + if let max_skips = max_skips { + __kwargs["max_skips"] = RemotePythonValue(max_skips) + } + if let stop_n_inliers = stop_n_inliers { + __kwargs["stop_n_inliers"] = RemotePythonValue(stop_n_inliers) + } + if let stop_score = stop_score { + __kwargs["stop_score"] = RemotePythonValue(stop_score) + } + if let stop_probability = stop_probability { + __kwargs["stop_probability"] = RemotePythonValue(stop_probability) + } + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RANSACRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "RANSACRegressor", args: __args, kwargs: __kwargs) + } + return PoolRANSACRegressor(pool: pool, owned: __owned) + } + + public func TheilSenRegressor(fit_intercept: Bool? = nil, copy_X: Bool? = nil, max_subpopulation: Int? = nil, n_subsamples: Int? = nil, max_iter: Int? = nil, tol: Double? = nil, random_state: Int? = nil, n_jobs: Int? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolTheilSenRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let copy_X = copy_X { + __kwargs["copy_X"] = RemotePythonValue(copy_X) + } + if let max_subpopulation = max_subpopulation { + __kwargs["max_subpopulation"] = RemotePythonValue(max_subpopulation) + } + if let n_subsamples = n_subsamples { + __kwargs["n_subsamples"] = RemotePythonValue(n_subsamples) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "TheilSenRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "TheilSenRegressor", args: __args, kwargs: __kwargs) + } + return PoolTheilSenRegressor(pool: pool, owned: __owned) + } + + public func OrthogonalMatchingPursuit(n_nonzero_coefs: Int? = nil, tol: Double? = nil, fit_intercept: Bool? = nil, precompute: String? = nil, worker: Int? = nil) async throws -> PoolOrthogonalMatchingPursuit { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_nonzero_coefs = n_nonzero_coefs { + __kwargs["n_nonzero_coefs"] = RemotePythonValue(n_nonzero_coefs) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let precompute = precompute { + __kwargs["precompute"] = RemotePythonValue(precompute) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "OrthogonalMatchingPursuit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "OrthogonalMatchingPursuit", args: __args, kwargs: __kwargs) + } + return PoolOrthogonalMatchingPursuit(pool: pool, owned: __owned) + } + + public func OrthogonalMatchingPursuitCV(copy: Bool? = nil, fit_intercept: Bool? = nil, max_iter: Int? = nil, cv: Int? = nil, n_jobs: Int? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolOrthogonalMatchingPursuitCV { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "OrthogonalMatchingPursuitCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "OrthogonalMatchingPursuitCV", args: __args, kwargs: __kwargs) + } + return PoolOrthogonalMatchingPursuitCV(pool: pool, owned: __owned) + } + + public func TweedieRegressor(power: Double? = nil, alpha: Double? = nil, fit_intercept: Bool? = nil, link: String? = nil, solver: String? = nil, max_iter: Int? = nil, tol: Double? = nil, warm_start: Bool? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolTweedieRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let power = power { + __kwargs["power"] = RemotePythonValue(power) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let link = link { + __kwargs["link"] = RemotePythonValue(link) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "TweedieRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "TweedieRegressor", args: __args, kwargs: __kwargs) + } + return PoolTweedieRegressor(pool: pool, owned: __owned) + } + + public func PoissonRegressor(alpha: Double? = nil, fit_intercept: Bool? = nil, solver: String? = nil, max_iter: Int? = nil, tol: Double? = nil, warm_start: Bool? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolPoissonRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "PoissonRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "PoissonRegressor", args: __args, kwargs: __kwargs) + } + return PoolPoissonRegressor(pool: pool, owned: __owned) + } + + public func GammaRegressor(alpha: Double? = nil, fit_intercept: Bool? = nil, solver: String? = nil, max_iter: Int? = nil, tol: Double? = nil, warm_start: Bool? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolGammaRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let solver = solver { + __kwargs["solver"] = RemotePythonValue(solver) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let warm_start = warm_start { + __kwargs["warm_start"] = RemotePythonValue(warm_start) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "GammaRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.linear_model", function: "GammaRegressor", args: __args, kwargs: __kwargs) + } + return PoolGammaRegressor(pool: pool, owned: __owned) + } +} + + +public struct PoolLinearRegression: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLinearRegression { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLinearRegression(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRidge: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolRidge { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRidge(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRidgeCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolRidgeCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRidgeCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRidgeClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolRidgeClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRidgeClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRidgeClassifierCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolRidgeClassifierCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRidgeClassifierCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLasso: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLasso { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLasso(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLassoCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLassoCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLassoCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLassoLars: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLassoLars { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLassoLars(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLassoLarsCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLassoLarsCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLassoLarsCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLassoLarsIC: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLassoLarsIC { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLassoLarsIC(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolElasticNet: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolElasticNet { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolElasticNet(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolElasticNetCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolElasticNetCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolElasticNetCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolMultiTaskLasso: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> PoolMultiTaskLasso { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMultiTaskLasso(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolMultiTaskElasticNet: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> PoolMultiTaskElasticNet { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMultiTaskElasticNet(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLars: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLars { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLars(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLarsCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLarsCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLarsCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLogisticRegression: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolLogisticRegression { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLogisticRegression(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLogisticRegressionCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolLogisticRegressionCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLogisticRegressionCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolSGDClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolSGDClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSGDClassifier(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolSGDClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolSGDClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolSGDRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolSGDRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSGDRegressor(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolSGDRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolSGDRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolSGDOneClassSVM: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSGDOneClassSVM { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSGDOneClassSVM(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSGDOneClassSVM { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolSGDOneClassSVM(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolPerceptron: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolPerceptron { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPerceptron(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolPerceptron { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolPerceptron(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolPassiveAggressiveClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolPassiveAggressiveClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPassiveAggressiveClassifier(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolPassiveAggressiveClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolPassiveAggressiveClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolPassiveAggressiveRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolPassiveAggressiveRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPassiveAggressiveRegressor(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolPassiveAggressiveRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolPassiveAggressiveRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolBayesianRidge: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolBayesianRidge { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBayesianRidge(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, return_std: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let return_std = return_std { + __kwargs["return_std"] = RemotePythonValue(return_std) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolARDRegression: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolARDRegression { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolARDRegression(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, return_std: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let return_std = return_std { + __kwargs["return_std"] = RemotePythonValue(return_std) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolHuberRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolHuberRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolHuberRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolQuantileRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolQuantileRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolQuantileRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRANSACRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolRANSACRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRANSACRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolTheilSenRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolTheilSenRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolTheilSenRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolOrthogonalMatchingPursuit: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolOrthogonalMatchingPursuit { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolOrthogonalMatchingPursuit(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolOrthogonalMatchingPursuitCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolOrthogonalMatchingPursuitCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolOrthogonalMatchingPursuitCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolTweedieRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolTweedieRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolTweedieRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolPoissonRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolPoissonRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPoissonRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolGammaRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolGammaRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolGammaRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + +// MARK: - Sklearn.Metrics + +extension PoolSklearn { + /// Pool-bound `sklearn.metrics` namespace. Calls run on this pool. + public var metrics: PoolSklearnMetrics { PoolSklearnMetrics(pool: pool) } +} + +public struct PoolSklearnMetrics: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func accuracy_score(y_true: [Int], y_pred: [Int], normalize: Bool? = nil, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "accuracy_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "accuracy_score", args: __args, kwargs: __kwargs) + } + + public func balanced_accuracy_score(y_true: [Int], y_pred: [Int], sample_weight: [Double]? = nil, adjusted: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let adjusted = adjusted { + __kwargs["adjusted"] = RemotePythonValue(adjusted) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "balanced_accuracy_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "balanced_accuracy_score", args: __args, kwargs: __kwargs) + } + + public func precision_score(y_true: [Int], y_pred: [Int], labels: [Int]? = nil, pos_label: Int? = nil, average: String? = nil, sample_weight: [Double]? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "precision_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "precision_score", args: __args, kwargs: __kwargs) + } + + public func recall_score(y_true: [Int], y_pred: [Int], labels: [Int]? = nil, pos_label: Int? = nil, average: String? = nil, sample_weight: [Double]? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "recall_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "recall_score", args: __args, kwargs: __kwargs) + } + + public func f1_score(y_true: [Int], y_pred: [Int], labels: [Int]? = nil, pos_label: Int? = nil, average: String? = nil, sample_weight: [Double]? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "f1_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "f1_score", args: __args, kwargs: __kwargs) + } + + public func fbeta_score(y_true: [Int], y_pred: [Int], beta: Double, labels: [Int]? = nil, pos_label: Int? = nil, average: String? = nil, sample_weight: [Double]? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + __args.append(RemotePythonValue(beta)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "fbeta_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "fbeta_score", args: __args, kwargs: __kwargs) + } + + public func precision_recall_fscore_support(y_true: [Int], y_pred: [Int], beta: Double? = nil, labels: [Int]? = nil, pos_label: Int? = nil, average: String? = nil, warn_for: (any PythonConvertible)? = nil, sample_weight: [Double]? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let beta = beta { + __kwargs["beta"] = RemotePythonValue(beta) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let warn_for = warn_for { + __kwargs["warn_for"] = RemotePythonValue(warn_for) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "precision_recall_fscore_support", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "precision_recall_fscore_support", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func classification_report(y_true: [Int], y_pred: [Int], labels: [Int]? = nil, target_names: [String]? = nil, sample_weight: [Double]? = nil, digits: Int? = nil, output_dict: Bool? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let target_names = target_names { + __kwargs["target_names"] = RemotePythonValue(target_names) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let digits = digits { + __kwargs["digits"] = RemotePythonValue(digits) + } + if let output_dict = output_dict { + __kwargs["output_dict"] = RemotePythonValue(output_dict) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "classification_report", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "classification_report", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func confusion_matrix(y_true: [Int], y_pred: [Int], labels: [Int]? = nil, sample_weight: [Double]? = nil, normalize: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "confusion_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "confusion_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func multilabel_confusion_matrix(y_true: [Int], y_pred: [Int], sample_weight: [Double]? = nil, labels: [Int]? = nil, samplewise: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let samplewise = samplewise { + __kwargs["samplewise"] = RemotePythonValue(samplewise) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "multilabel_confusion_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "multilabel_confusion_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func roc_auc_score(y_true: [Int], y_score: any PythonConvertible, average: String? = nil, sample_weight: [Double]? = nil, max_fpr: Double? = nil, multi_class: String? = nil, labels: [Int]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let max_fpr = max_fpr { + __kwargs["max_fpr"] = RemotePythonValue(max_fpr) + } + if let multi_class = multi_class { + __kwargs["multi_class"] = RemotePythonValue(multi_class) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "roc_auc_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "roc_auc_score", args: __args, kwargs: __kwargs) + } + + public func roc_curve(y_true: [Int], y_score: [Double], pos_label: Int? = nil, sample_weight: [Double]? = nil, drop_intermediate: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let drop_intermediate = drop_intermediate { + __kwargs["drop_intermediate"] = RemotePythonValue(drop_intermediate) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "roc_curve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "roc_curve", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func auc(x: [Double], y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(x)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "auc", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "auc", args: __args, kwargs: __kwargs) + } + + public func average_precision_score(y_true: [Int], y_score: [Double], average: String? = nil, pos_label: Int? = nil, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "average_precision_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "average_precision_score", args: __args, kwargs: __kwargs) + } + + public func precision_recall_curve(y_true: [Int], y_score: [Double], pos_label: Int? = nil, sample_weight: [Double]? = nil, drop_intermediate: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let drop_intermediate = drop_intermediate { + __kwargs["drop_intermediate"] = RemotePythonValue(drop_intermediate) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "precision_recall_curve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "precision_recall_curve", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func det_curve(y_true: [Int], y_score: [Double], pos_label: Int? = nil, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "det_curve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "det_curve", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func log_loss(y_true: [Int], y_pred: any PythonConvertible, eps: String? = nil, normalize: Bool? = nil, sample_weight: [Double]? = nil, labels: [Int]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "log_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "log_loss", args: __args, kwargs: __kwargs) + } + + public func brier_score_loss(y_true: [Int], y_prob: [Double], sample_weight: [Double]? = nil, pos_label: Int? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_prob)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "brier_score_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "brier_score_loss", args: __args, kwargs: __kwargs) + } + + public func hinge_loss(y_true: [Int], pred_decision: [Double], labels: [Int]? = nil, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(pred_decision)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "hinge_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "hinge_loss", args: __args, kwargs: __kwargs) + } + + public func jaccard_score(y_true: [Int], y_pred: [Int], labels: [Int]? = nil, pos_label: Int? = nil, average: String? = nil, sample_weight: [Double]? = nil, zero_division: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let average = average { + __kwargs["average"] = RemotePythonValue(average) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let zero_division = zero_division { + __kwargs["zero_division"] = RemotePythonValue(zero_division) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "jaccard_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "jaccard_score", args: __args, kwargs: __kwargs) + } + + public func matthews_corrcoef(y_true: [Int], y_pred: [Int], sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "matthews_corrcoef", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "matthews_corrcoef", args: __args, kwargs: __kwargs) + } + + public func cohen_kappa_score(y1: [Int], y2: [Int], labels: [Int]? = nil, weights: String? = nil, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y1)) + __args.append(RemotePythonValue(y2)) + var __kwargs: [String: RemotePythonValue] = [:] + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "cohen_kappa_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "cohen_kappa_score", args: __args, kwargs: __kwargs) + } + + public func hamming_loss(y_true: [Int], y_pred: [Int], sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "hamming_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "hamming_loss", args: __args, kwargs: __kwargs) + } + + public func zero_one_loss(y_true: [Int], y_pred: [Int], normalize: Bool? = nil, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "zero_one_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "zero_one_loss", args: __args, kwargs: __kwargs) + } + + public func top_k_accuracy_score(y_true: [Int], y_score: any PythonConvertible, k: Int? = nil, normalize: Bool? = nil, sample_weight: [Double]? = nil, labels: [Int]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let normalize = normalize { + __kwargs["normalize"] = RemotePythonValue(normalize) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "top_k_accuracy_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "top_k_accuracy_score", args: __args, kwargs: __kwargs) + } + + public func d2_log_loss_score(y_true: [Int], y_pred: any PythonConvertible, sample_weight: [Double]? = nil, labels: [Int]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let labels = labels { + __kwargs["labels"] = RemotePythonValue(labels) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_log_loss_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_log_loss_score", args: __args, kwargs: __kwargs) + } + + public func mean_squared_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, squared: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let squared = squared { + __kwargs["squared"] = RemotePythonValue(squared) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_squared_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_squared_error", args: __args, kwargs: __kwargs) + } + + public func mean_absolute_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_absolute_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_absolute_error", args: __args, kwargs: __kwargs) + } + + public func mean_absolute_percentage_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_absolute_percentage_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_absolute_percentage_error", args: __args, kwargs: __kwargs) + } + + public func median_absolute_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "median_absolute_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "median_absolute_error", args: __args, kwargs: __kwargs) + } + + public func r2_score(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, force_finite: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let force_finite = force_finite { + __kwargs["force_finite"] = RemotePythonValue(force_finite) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "r2_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "r2_score", args: __args, kwargs: __kwargs) + } + + public func explained_variance_score(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, force_finite: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let force_finite = force_finite { + __kwargs["force_finite"] = RemotePythonValue(force_finite) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "explained_variance_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "explained_variance_score", args: __args, kwargs: __kwargs) + } + + public func max_error(y_true: [Double], y_pred: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "max_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "max_error", args: __args, kwargs: __kwargs) + } + + public func mean_squared_log_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, squared: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let squared = squared { + __kwargs["squared"] = RemotePythonValue(squared) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_squared_log_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_squared_log_error", args: __args, kwargs: __kwargs) + } + + public func root_mean_squared_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "root_mean_squared_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "root_mean_squared_error", args: __args, kwargs: __kwargs) + } + + public func root_mean_squared_log_error(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "root_mean_squared_log_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "root_mean_squared_log_error", args: __args, kwargs: __kwargs) + } + + public func mean_poisson_deviance(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_poisson_deviance", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_poisson_deviance", args: __args, kwargs: __kwargs) + } + + public func mean_gamma_deviance(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_gamma_deviance", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_gamma_deviance", args: __args, kwargs: __kwargs) + } + + public func mean_tweedie_deviance(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, power: Double? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let power = power { + __kwargs["power"] = RemotePythonValue(power) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_tweedie_deviance", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_tweedie_deviance", args: __args, kwargs: __kwargs) + } + + public func d2_tweedie_score(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, power: Double? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let power = power { + __kwargs["power"] = RemotePythonValue(power) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_tweedie_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_tweedie_score", args: __args, kwargs: __kwargs) + } + + public func d2_pinball_score(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, alpha: Double? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_pinball_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_pinball_score", args: __args, kwargs: __kwargs) + } + + public func d2_absolute_error_score(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_absolute_error_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "d2_absolute_error_score", args: __args, kwargs: __kwargs) + } + + public func mean_pinball_loss(y_true: [Double], y_pred: [Double], sample_weight: [Double]? = nil, alpha: Double? = nil, multioutput: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let multioutput = multioutput { + __kwargs["multioutput"] = RemotePythonValue(multioutput) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_pinball_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mean_pinball_loss", args: __args, kwargs: __kwargs) + } + + public func adjusted_rand_score(labels_true: [Int], labels_pred: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "adjusted_rand_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "adjusted_rand_score", args: __args, kwargs: __kwargs) + } + + public func rand_score(labels_true: [Int], labels_pred: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "rand_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "rand_score", args: __args, kwargs: __kwargs) + } + + public func adjusted_mutual_info_score(labels_true: [Int], labels_pred: [Int], average_method: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let average_method = average_method { + __kwargs["average_method"] = RemotePythonValue(average_method) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "adjusted_mutual_info_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "adjusted_mutual_info_score", args: __args, kwargs: __kwargs) + } + + public func mutual_info_score(labels_true: [Int], labels_pred: [Int], contingency: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + if let contingency = contingency { + __args.append(RemotePythonValue(contingency)) + } + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "mutual_info_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "mutual_info_score", args: __args, kwargs: __kwargs) + } + + public func normalized_mutual_info_score(labels_true: [Int], labels_pred: [Int], average_method: String? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let average_method = average_method { + __kwargs["average_method"] = RemotePythonValue(average_method) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "normalized_mutual_info_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "normalized_mutual_info_score", args: __args, kwargs: __kwargs) + } + + public func homogeneity_score(labels_true: [Int], labels_pred: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "homogeneity_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "homogeneity_score", args: __args, kwargs: __kwargs) + } + + public func completeness_score(labels_true: [Int], labels_pred: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "completeness_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "completeness_score", args: __args, kwargs: __kwargs) + } + + public func v_measure_score(labels_true: [Int], labels_pred: [Int], beta: Double? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let beta = beta { + __kwargs["beta"] = RemotePythonValue(beta) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "v_measure_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "v_measure_score", args: __args, kwargs: __kwargs) + } + + public func homogeneity_completeness_v_measure(labels_true: [Int], labels_pred: [Int], beta: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let beta = beta { + __kwargs["beta"] = RemotePythonValue(beta) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "homogeneity_completeness_v_measure", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "homogeneity_completeness_v_measure", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fowlkes_mallows_score(labels_true: [Int], labels_pred: [Int], sparse: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sparse = sparse { + __kwargs["sparse"] = RemotePythonValue(sparse) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "fowlkes_mallows_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "fowlkes_mallows_score", args: __args, kwargs: __kwargs) + } + + public func silhouette_score(X: any PythonConvertible, labels: [Int], metric: String? = nil, sample_size: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(labels)) + var __kwargs: [String: RemotePythonValue] = [:] + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let sample_size = sample_size { + __kwargs["sample_size"] = RemotePythonValue(sample_size) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "silhouette_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "silhouette_score", args: __args, kwargs: __kwargs) + } + + public func silhouette_samples(X: any PythonConvertible, labels: [Int], metric: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(labels)) + var __kwargs: [String: RemotePythonValue] = [:] + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "silhouette_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "silhouette_samples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func calinski_harabasz_score(X: any PythonConvertible, labels: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(labels)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "calinski_harabasz_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "calinski_harabasz_score", args: __args, kwargs: __kwargs) + } + + public func davies_bouldin_score(X: any PythonConvertible, labels: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(labels)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "davies_bouldin_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "davies_bouldin_score", args: __args, kwargs: __kwargs) + } + + public func contingency_matrix(labels_true: [Int], labels_pred: [Int], eps: Double? = nil, sparse: Bool? = nil, dtype: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + var __kwargs: [String: RemotePythonValue] = [:] + if let eps = eps { + __kwargs["eps"] = RemotePythonValue(eps) + } + if let sparse = sparse { + __kwargs["sparse"] = RemotePythonValue(sparse) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "contingency_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "contingency_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func pair_confusion_matrix(labels_true: [Int], labels_pred: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(labels_true)) + __args.append(RemotePythonValue(labels_pred)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "pair_confusion_matrix", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "pair_confusion_matrix", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func pairwise_distances(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, metric: String? = nil, n_jobs: Int? = nil, force_all_finite: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let force_all_finite = force_all_finite { + __kwargs["force_all_finite"] = RemotePythonValue(force_all_finite) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "pairwise_distances", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "pairwise_distances", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func pairwise_kernels(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, metric: String? = nil, filter_params: Bool? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let filter_params = filter_params { + __kwargs["filter_params"] = RemotePythonValue(filter_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "pairwise_kernels", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "pairwise_kernels", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func euclidean_distances(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, Y_norm_squared: (any PythonConvertible)? = nil, squared: Bool? = nil, X_norm_squared: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + if let Y_norm_squared = Y_norm_squared { + __args.append(RemotePythonValue(Y_norm_squared)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let squared = squared { + __kwargs["squared"] = RemotePythonValue(squared) + } + if let X_norm_squared = X_norm_squared { + __kwargs["X_norm_squared"] = RemotePythonValue(X_norm_squared) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "euclidean_distances", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "euclidean_distances", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func manhattan_distances(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, sum_over_features: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let sum_over_features = sum_over_features { + __kwargs["sum_over_features"] = RemotePythonValue(sum_over_features) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "manhattan_distances", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "manhattan_distances", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cosine_distances(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "cosine_distances", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "cosine_distances", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cosine_similarity(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, dense_output: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dense_output = dense_output { + __kwargs["dense_output"] = RemotePythonValue(dense_output) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "cosine_similarity", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "cosine_similarity", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func linear_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, dense_output: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let dense_output = dense_output { + __kwargs["dense_output"] = RemotePythonValue(dense_output) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "linear_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "linear_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func polynomial_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, degree: Int? = nil, gamma: Double? = nil, coef0: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "polynomial_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "polynomial_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func sigmoid_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, gamma: Double? = nil, coef0: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "sigmoid_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "sigmoid_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func rbf_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, gamma: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "rbf_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "rbf_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func laplacian_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, gamma: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "laplacian_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "laplacian_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func chi2_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, gamma: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "chi2_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "chi2_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func additive_chi2_kernel(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "additive_chi2_kernel", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "additive_chi2_kernel", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func haversine_distances(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "haversine_distances", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "haversine_distances", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func nan_euclidean_distances(X: any PythonConvertible, Y: (any PythonConvertible)? = nil, squared: Bool? = nil, missing_values: Double? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let Y = Y { + __args.append(RemotePythonValue(Y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let squared = squared { + __kwargs["squared"] = RemotePythonValue(squared) + } + if let missing_values = missing_values { + __kwargs["missing_values"] = RemotePythonValue(missing_values) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "nan_euclidean_distances", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "nan_euclidean_distances", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func ndcg_score(y_true: any PythonConvertible, y_score: any PythonConvertible, k: Int? = nil, sample_weight: [Double]? = nil, ignore_ties: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let ignore_ties = ignore_ties { + __kwargs["ignore_ties"] = RemotePythonValue(ignore_ties) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "ndcg_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "ndcg_score", args: __args, kwargs: __kwargs) + } + + public func dcg_score(y_true: any PythonConvertible, y_score: any PythonConvertible, k: Int? = nil, log_base: Double? = nil, sample_weight: [Double]? = nil, ignore_ties: Bool? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let log_base = log_base { + __kwargs["log_base"] = RemotePythonValue(log_base) + } + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let ignore_ties = ignore_ties { + __kwargs["ignore_ties"] = RemotePythonValue(ignore_ties) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "dcg_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "dcg_score", args: __args, kwargs: __kwargs) + } + + public func label_ranking_average_precision_score(y_true: any PythonConvertible, y_score: any PythonConvertible, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "label_ranking_average_precision_score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "label_ranking_average_precision_score", args: __args, kwargs: __kwargs) + } + + public func label_ranking_loss(y_true: any PythonConvertible, y_score: any PythonConvertible, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "label_ranking_loss", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "label_ranking_loss", args: __args, kwargs: __kwargs) + } + + public func coverage_error(y_true: any PythonConvertible, y_score: any PythonConvertible, sample_weight: [Double]? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y_true)) + __args.append(RemotePythonValue(y_score)) + var __kwargs: [String: RemotePythonValue] = [:] + if let sample_weight = sample_weight { + __kwargs["sample_weight"] = RemotePythonValue(sample_weight) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "coverage_error", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "coverage_error", args: __args, kwargs: __kwargs) + } + + public func make_scorer(score_func: any PythonConvertible, greater_is_better: Bool? = nil, needs_proba: Bool? = nil, needs_threshold: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(score_func)) + var __kwargs: [String: RemotePythonValue] = [:] + if let greater_is_better = greater_is_better { + __kwargs["greater_is_better"] = RemotePythonValue(greater_is_better) + } + if let needs_proba = needs_proba { + __kwargs["needs_proba"] = RemotePythonValue(needs_proba) + } + if let needs_threshold = needs_threshold { + __kwargs["needs_threshold"] = RemotePythonValue(needs_threshold) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "make_scorer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "make_scorer", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_scorer(scoring: String, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(scoring)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "get_scorer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "get_scorer", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_scorer_names(worker: Int? = nil) async throws -> [String] { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.invokeResult(module: "sklearn.metrics", function: "get_scorer_names", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.metrics", function: "get_scorer_names", args: __args, kwargs: __kwargs) + } + + public func check_scoring(estimator: any PythonConvertible, scoring: String? = nil, allow_none: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + var __kwargs: [String: RemotePythonValue] = [:] + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let allow_none = allow_none { + __kwargs["allow_none"] = RemotePythonValue(allow_none) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "check_scoring", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "check_scoring", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func ConfusionMatrixDisplay(confusion_matrix: any PythonConvertible, display_labels: [String]? = nil, worker: Int? = nil) async throws -> PoolConfusionMatrixDisplay { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(confusion_matrix)) + var __kwargs: [String: RemotePythonValue] = [:] + if let display_labels = display_labels { + __kwargs["display_labels"] = RemotePythonValue(display_labels) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "ConfusionMatrixDisplay", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "ConfusionMatrixDisplay", args: __args, kwargs: __kwargs) + } + return PoolConfusionMatrixDisplay(pool: pool, owned: __owned) + } + + public func RocCurveDisplay(fpr: any PythonConvertible, tpr: any PythonConvertible, roc_auc: Double? = nil, estimator_name: String? = nil, pos_label: Int? = nil, worker: Int? = nil) async throws -> PoolRocCurveDisplay { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(fpr)) + __args.append(RemotePythonValue(tpr)) + var __kwargs: [String: RemotePythonValue] = [:] + if let roc_auc = roc_auc { + __kwargs["roc_auc"] = RemotePythonValue(roc_auc) + } + if let estimator_name = estimator_name { + __kwargs["estimator_name"] = RemotePythonValue(estimator_name) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "RocCurveDisplay", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "RocCurveDisplay", args: __args, kwargs: __kwargs) + } + return PoolRocCurveDisplay(pool: pool, owned: __owned) + } + + public func PrecisionRecallDisplay(precision: any PythonConvertible, recall: any PythonConvertible, average_precision: Double? = nil, estimator_name: String? = nil, pos_label: Int? = nil, prevalence_pos_label: Double? = nil, worker: Int? = nil) async throws -> PoolPrecisionRecallDisplay { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(precision)) + __args.append(RemotePythonValue(recall)) + var __kwargs: [String: RemotePythonValue] = [:] + if let average_precision = average_precision { + __kwargs["average_precision"] = RemotePythonValue(average_precision) + } + if let estimator_name = estimator_name { + __kwargs["estimator_name"] = RemotePythonValue(estimator_name) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let prevalence_pos_label = prevalence_pos_label { + __kwargs["prevalence_pos_label"] = RemotePythonValue(prevalence_pos_label) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "PrecisionRecallDisplay", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "PrecisionRecallDisplay", args: __args, kwargs: __kwargs) + } + return PoolPrecisionRecallDisplay(pool: pool, owned: __owned) + } + + public func DetCurveDisplay(fpr: any PythonConvertible, fnr: any PythonConvertible, estimator_name: String? = nil, pos_label: Int? = nil, worker: Int? = nil) async throws -> PoolDetCurveDisplay { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(fpr)) + __args.append(RemotePythonValue(fnr)) + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator_name = estimator_name { + __kwargs["estimator_name"] = RemotePythonValue(estimator_name) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "DetCurveDisplay", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "DetCurveDisplay", args: __args, kwargs: __kwargs) + } + return PoolDetCurveDisplay(pool: pool, owned: __owned) + } + + public func CalibrationDisplay(prob_true: any PythonConvertible, prob_pred: any PythonConvertible, y_prob: any PythonConvertible, estimator_name: String? = nil, pos_label: Int? = nil, worker: Int? = nil) async throws -> PoolCalibrationDisplay { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(prob_true)) + __args.append(RemotePythonValue(prob_pred)) + __args.append(RemotePythonValue(y_prob)) + var __kwargs: [String: RemotePythonValue] = [:] + if let estimator_name = estimator_name { + __kwargs["estimator_name"] = RemotePythonValue(estimator_name) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "CalibrationDisplay", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.metrics", function: "CalibrationDisplay", args: __args, kwargs: __kwargs) + } + return PoolCalibrationDisplay(pool: pool, owned: __owned) + } +} + + +public struct PoolConfusionMatrixDisplay: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func plot(include_values: Bool? = nil, cmap: String? = nil, xticks_rotation: String? = nil, values_format: String? = nil, ax: (any PythonConvertible)? = nil, colorbar: Bool? = nil, im_kw: (any PythonConvertible)? = nil, text_kw: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolConfusionMatrixDisplay { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let include_values = include_values { + __kwargs["include_values"] = RemotePythonValue(include_values) + } + if let cmap = cmap { + __kwargs["cmap"] = RemotePythonValue(cmap) + } + if let xticks_rotation = xticks_rotation { + __kwargs["xticks_rotation"] = RemotePythonValue(xticks_rotation) + } + if let values_format = values_format { + __kwargs["values_format"] = RemotePythonValue(values_format) + } + if let ax = ax { + __kwargs["ax"] = RemotePythonValue(ax) + } + if let colorbar = colorbar { + __kwargs["colorbar"] = RemotePythonValue(colorbar) + } + if let im_kw = im_kw { + __kwargs["im_kw"] = RemotePythonValue(im_kw) + } + if let text_kw = text_kw { + __kwargs["text_kw"] = RemotePythonValue(text_kw) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs) + } + return PoolConfusionMatrixDisplay(pool: pool, owned: __owned) + } +} + + +public struct PoolRocCurveDisplay: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func plot(ax: (any PythonConvertible)? = nil, name: String? = nil, plot_chance_level: Bool? = nil, chance_level_kw: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolRocCurveDisplay { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ax = ax { + __kwargs["ax"] = RemotePythonValue(ax) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let plot_chance_level = plot_chance_level { + __kwargs["plot_chance_level"] = RemotePythonValue(plot_chance_level) + } + if let chance_level_kw = chance_level_kw { + __kwargs["chance_level_kw"] = RemotePythonValue(chance_level_kw) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs) + } + return PoolRocCurveDisplay(pool: pool, owned: __owned) + } +} + + +public struct PoolPrecisionRecallDisplay: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func plot(ax: (any PythonConvertible)? = nil, name: String? = nil, plot_chance_level: Bool? = nil, chance_level_kw: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolPrecisionRecallDisplay { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ax = ax { + __kwargs["ax"] = RemotePythonValue(ax) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let plot_chance_level = plot_chance_level { + __kwargs["plot_chance_level"] = RemotePythonValue(plot_chance_level) + } + if let chance_level_kw = chance_level_kw { + __kwargs["chance_level_kw"] = RemotePythonValue(chance_level_kw) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs) + } + return PoolPrecisionRecallDisplay(pool: pool, owned: __owned) + } +} + + +public struct PoolDetCurveDisplay: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func plot(ax: (any PythonConvertible)? = nil, name: String? = nil, worker: Int? = nil) async throws -> PoolDetCurveDisplay { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ax = ax { + __kwargs["ax"] = RemotePythonValue(ax) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs) + } + return PoolDetCurveDisplay(pool: pool, owned: __owned) + } +} + + +public struct PoolCalibrationDisplay: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func plot(ax: (any PythonConvertible)? = nil, name: String? = nil, ref_line: Bool? = nil, worker: Int? = nil) async throws -> PoolCalibrationDisplay { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let ax = ax { + __kwargs["ax"] = RemotePythonValue(ax) + } + if let name = name { + __kwargs["name"] = RemotePythonValue(name) + } + if let ref_line = ref_line { + __kwargs["ref_line"] = RemotePythonValue(ref_line) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "plot", args: __args, kwargs: __kwargs) + } + return PoolCalibrationDisplay(pool: pool, owned: __owned) + } +} + +// MARK: - Sklearn.ModelSelection + +extension PoolSklearn { + /// Pool-bound `sklearn.model_selection` namespace. Calls run on this pool. + public var modelselection: PoolSklearnModelselection { PoolSklearnModelselection(pool: pool) } +} + +public struct PoolSklearnModelselection: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func train_test_split(X: any PythonConvertible, y: (any PythonConvertible)? = nil, test_size: Double? = nil, train_size: Double? = nil, random_state: Int? = nil, shuffle: Bool? = nil, stratify: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let y = y { + __args.append(RemotePythonValue(y)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let test_size = test_size { + __kwargs["test_size"] = RemotePythonValue(test_size) + } + if let train_size = train_size { + __kwargs["train_size"] = RemotePythonValue(train_size) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let stratify = stratify { + __kwargs["stratify"] = RemotePythonValue(stratify) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "train_test_split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "train_test_split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cross_val_score(estimator: any PythonConvertible, X: any PythonConvertible, y: (any PythonConvertible)? = nil, groups: (any PythonConvertible)? = nil, scoring: String? = nil, cv: Int? = nil, n_jobs: Int? = nil, verbose: Int? = nil, fit_params: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(X)) + if let y = y { + __args.append(RemotePythonValue(y)) + } + if let groups = groups { + __args.append(RemotePythonValue(groups)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let fit_params = fit_params { + __kwargs["fit_params"] = RemotePythonValue(fit_params) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "cross_val_score", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "cross_val_score", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cross_val_predict(estimator: any PythonConvertible, X: any PythonConvertible, y: (any PythonConvertible)? = nil, groups: (any PythonConvertible)? = nil, cv: Int? = nil, n_jobs: Int? = nil, verbose: Int? = nil, fit_params: (any PythonConvertible)? = nil, method: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(X)) + if let y = y { + __args.append(RemotePythonValue(y)) + } + if let groups = groups { + __args.append(RemotePythonValue(groups)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let fit_params = fit_params { + __kwargs["fit_params"] = RemotePythonValue(fit_params) + } + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "cross_val_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "cross_val_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func cross_validate(estimator: any PythonConvertible, X: any PythonConvertible, y: (any PythonConvertible)? = nil, groups: (any PythonConvertible)? = nil, scoring: (any PythonConvertible)? = nil, cv: Int? = nil, n_jobs: Int? = nil, verbose: Int? = nil, fit_params: (any PythonConvertible)? = nil, return_train_score: Bool? = nil, return_estimator: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(X)) + if let y = y { + __args.append(RemotePythonValue(y)) + } + if let groups = groups { + __args.append(RemotePythonValue(groups)) + } + if let scoring = scoring { + __args.append(RemotePythonValue(scoring)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let fit_params = fit_params { + __kwargs["fit_params"] = RemotePythonValue(fit_params) + } + if let return_train_score = return_train_score { + __kwargs["return_train_score"] = RemotePythonValue(return_train_score) + } + if let return_estimator = return_estimator { + __kwargs["return_estimator"] = RemotePythonValue(return_estimator) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "cross_validate", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "cross_validate", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func learning_curve(estimator: any PythonConvertible, X: any PythonConvertible, y: any PythonConvertible, groups: (any PythonConvertible)? = nil, train_sizes: (any PythonConvertible)? = nil, cv: Int? = nil, scoring: String? = nil, n_jobs: Int? = nil, verbose: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, return_times: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + if let groups = groups { + __args.append(RemotePythonValue(groups)) + } + if let train_sizes = train_sizes { + __args.append(RemotePythonValue(train_sizes)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let return_times = return_times { + __kwargs["return_times"] = RemotePythonValue(return_times) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "learning_curve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "learning_curve", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func validation_curve(estimator: any PythonConvertible, X: any PythonConvertible, y: any PythonConvertible, param_name: String, param_range: any PythonConvertible, groups: (any PythonConvertible)? = nil, cv: Int? = nil, scoring: String? = nil, n_jobs: Int? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + __args.append(RemotePythonValue(param_name)) + __args.append(RemotePythonValue(param_range)) + if let groups = groups { + __args.append(RemotePythonValue(groups)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "validation_curve", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "validation_curve", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func permutation_test_score(estimator: any PythonConvertible, X: any PythonConvertible, y: any PythonConvertible, groups: (any PythonConvertible)? = nil, cv: Int? = nil, n_permutations: Int? = nil, n_jobs: Int? = nil, random_state: Int? = nil, verbose: Int? = nil, scoring: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + if let groups = groups { + __args.append(RemotePythonValue(groups)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let n_permutations = n_permutations { + __kwargs["n_permutations"] = RemotePythonValue(n_permutations) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "permutation_test_score", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "permutation_test_score", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func KFold(n_splits: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolKFold { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "KFold", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "KFold", args: __args, kwargs: __kwargs) + } + return PoolKFold(pool: pool, owned: __owned) + } + + public func StratifiedKFold(n_splits: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolStratifiedKFold { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "StratifiedKFold", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "StratifiedKFold", args: __args, kwargs: __kwargs) + } + return PoolStratifiedKFold(pool: pool, owned: __owned) + } + + public func LeaveOneOut(worker: Int? = nil) async throws -> PoolLeaveOneOut { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "LeaveOneOut", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "LeaveOneOut", args: __args, kwargs: __kwargs) + } + return PoolLeaveOneOut(pool: pool, owned: __owned) + } + + public func LeavePOut(p: Int, worker: Int? = nil) async throws -> PoolLeavePOut { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(p)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "LeavePOut", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "LeavePOut", args: __args, kwargs: __kwargs) + } + return PoolLeavePOut(pool: pool, owned: __owned) + } + + public func ShuffleSplit(n_splits: Int? = nil, test_size: Double? = nil, train_size: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolShuffleSplit { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let test_size = test_size { + __kwargs["test_size"] = RemotePythonValue(test_size) + } + if let train_size = train_size { + __kwargs["train_size"] = RemotePythonValue(train_size) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "ShuffleSplit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "ShuffleSplit", args: __args, kwargs: __kwargs) + } + return PoolShuffleSplit(pool: pool, owned: __owned) + } + + public func StratifiedShuffleSplit(n_splits: Int? = nil, test_size: Double? = nil, train_size: Double? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolStratifiedShuffleSplit { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let test_size = test_size { + __kwargs["test_size"] = RemotePythonValue(test_size) + } + if let train_size = train_size { + __kwargs["train_size"] = RemotePythonValue(train_size) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "StratifiedShuffleSplit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "StratifiedShuffleSplit", args: __args, kwargs: __kwargs) + } + return PoolStratifiedShuffleSplit(pool: pool, owned: __owned) + } + + public func GroupKFold(n_splits: Int? = nil, worker: Int? = nil) async throws -> PoolGroupKFold { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "GroupKFold", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "GroupKFold", args: __args, kwargs: __kwargs) + } + return PoolGroupKFold(pool: pool, owned: __owned) + } + + public func TimeSeriesSplit(n_splits: Int? = nil, max_train_size: Int? = nil, test_size: Int? = nil, gap: Int? = nil, worker: Int? = nil) async throws -> PoolTimeSeriesSplit { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let max_train_size = max_train_size { + __kwargs["max_train_size"] = RemotePythonValue(max_train_size) + } + if let test_size = test_size { + __kwargs["test_size"] = RemotePythonValue(test_size) + } + if let gap = gap { + __kwargs["gap"] = RemotePythonValue(gap) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "TimeSeriesSplit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "TimeSeriesSplit", args: __args, kwargs: __kwargs) + } + return PoolTimeSeriesSplit(pool: pool, owned: __owned) + } + + public func RepeatedKFold(n_splits: Int? = nil, n_repeats: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolRepeatedKFold { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let n_repeats = n_repeats { + __kwargs["n_repeats"] = RemotePythonValue(n_repeats) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "RepeatedKFold", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "RepeatedKFold", args: __args, kwargs: __kwargs) + } + return PoolRepeatedKFold(pool: pool, owned: __owned) + } + + public func RepeatedStratifiedKFold(n_splits: Int? = nil, n_repeats: Int? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolRepeatedStratifiedKFold { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_splits = n_splits { + __kwargs["n_splits"] = RemotePythonValue(n_splits) + } + if let n_repeats = n_repeats { + __kwargs["n_repeats"] = RemotePythonValue(n_repeats) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "RepeatedStratifiedKFold", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "RepeatedStratifiedKFold", args: __args, kwargs: __kwargs) + } + return PoolRepeatedStratifiedKFold(pool: pool, owned: __owned) + } + + public func GridSearchCV(estimator: any PythonConvertible, param_grid: any PythonConvertible, scoring: String? = nil, n_jobs: Int? = nil, refit: Bool? = nil, cv: Int? = nil, verbose: Int? = nil, return_train_score: Bool? = nil, worker: Int? = nil) async throws -> PoolGridSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(param_grid)) + var __kwargs: [String: RemotePythonValue] = [:] + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let refit = refit { + __kwargs["refit"] = RemotePythonValue(refit) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let return_train_score = return_train_score { + __kwargs["return_train_score"] = RemotePythonValue(return_train_score) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "GridSearchCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "GridSearchCV", args: __args, kwargs: __kwargs) + } + return PoolGridSearchCV(pool: pool, owned: __owned) + } + + public func RandomizedSearchCV(estimator: any PythonConvertible, param_distributions: any PythonConvertible, n_iter: Int? = nil, scoring: String? = nil, n_jobs: Int? = nil, refit: Bool? = nil, cv: Int? = nil, verbose: Int? = nil, random_state: Int? = nil, return_train_score: Bool? = nil, worker: Int? = nil) async throws -> PoolRandomizedSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(param_distributions)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_iter = n_iter { + __kwargs["n_iter"] = RemotePythonValue(n_iter) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let refit = refit { + __kwargs["refit"] = RemotePythonValue(refit) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let return_train_score = return_train_score { + __kwargs["return_train_score"] = RemotePythonValue(return_train_score) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "RandomizedSearchCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "RandomizedSearchCV", args: __args, kwargs: __kwargs) + } + return PoolRandomizedSearchCV(pool: pool, owned: __owned) + } + + public func HalvingGridSearchCV(estimator: any PythonConvertible, param_grid: any PythonConvertible, factor: Int? = nil, resource: String? = nil, max_resources: String? = nil, min_resources: String? = nil, aggressive_elimination: Bool? = nil, cv: Int? = nil, scoring: String? = nil, refit: Bool? = nil, random_state: Int? = nil, n_jobs: Int? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolHalvingGridSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(param_grid)) + var __kwargs: [String: RemotePythonValue] = [:] + if let factor = factor { + __kwargs["factor"] = RemotePythonValue(factor) + } + if let resource = resource { + __kwargs["resource"] = RemotePythonValue(resource) + } + if let max_resources = max_resources { + __kwargs["max_resources"] = RemotePythonValue(max_resources) + } + if let min_resources = min_resources { + __kwargs["min_resources"] = RemotePythonValue(min_resources) + } + if let aggressive_elimination = aggressive_elimination { + __kwargs["aggressive_elimination"] = RemotePythonValue(aggressive_elimination) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let refit = refit { + __kwargs["refit"] = RemotePythonValue(refit) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "HalvingGridSearchCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "HalvingGridSearchCV", args: __args, kwargs: __kwargs) + } + return PoolHalvingGridSearchCV(pool: pool, owned: __owned) + } + + public func HalvingRandomSearchCV(estimator: any PythonConvertible, param_distributions: any PythonConvertible, n_candidates: String? = nil, factor: Int? = nil, resource: String? = nil, max_resources: String? = nil, min_resources: String? = nil, aggressive_elimination: Bool? = nil, cv: Int? = nil, scoring: String? = nil, refit: Bool? = nil, random_state: Int? = nil, n_jobs: Int? = nil, verbose: Int? = nil, worker: Int? = nil) async throws -> PoolHalvingRandomSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(estimator)) + __args.append(RemotePythonValue(param_distributions)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_candidates = n_candidates { + __kwargs["n_candidates"] = RemotePythonValue(n_candidates) + } + if let factor = factor { + __kwargs["factor"] = RemotePythonValue(factor) + } + if let resource = resource { + __kwargs["resource"] = RemotePythonValue(resource) + } + if let max_resources = max_resources { + __kwargs["max_resources"] = RemotePythonValue(max_resources) + } + if let min_resources = min_resources { + __kwargs["min_resources"] = RemotePythonValue(min_resources) + } + if let aggressive_elimination = aggressive_elimination { + __kwargs["aggressive_elimination"] = RemotePythonValue(aggressive_elimination) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let scoring = scoring { + __kwargs["scoring"] = RemotePythonValue(scoring) + } + if let refit = refit { + __kwargs["refit"] = RemotePythonValue(refit) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "HalvingRandomSearchCV", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "HalvingRandomSearchCV", args: __args, kwargs: __kwargs) + } + return PoolHalvingRandomSearchCV(pool: pool, owned: __owned) + } + + public func ParameterGrid(param_grid: any PythonConvertible, worker: Int? = nil) async throws -> PoolParameterGrid { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(param_grid)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "ParameterGrid", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "ParameterGrid", args: __args, kwargs: __kwargs) + } + return PoolParameterGrid(pool: pool, owned: __owned) + } + + public func ParameterSampler(param_distributions: any PythonConvertible, n_iter: Int, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolParameterSampler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(param_distributions)) + __args.append(RemotePythonValue(n_iter)) + var __kwargs: [String: RemotePythonValue] = [:] + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "ParameterSampler", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.model_selection", function: "ParameterSampler", args: __args, kwargs: __kwargs) + } + return PoolParameterSampler(pool: pool, owned: __owned) + } +} + + +public struct PoolKFold: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolStratifiedKFold: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLeaveOneOut: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: any PythonConvertible, worker: Int? = nil) async throws -> Int { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLeavePOut: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: any PythonConvertible, worker: Int? = nil) async throws -> Int { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolShuffleSplit: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolStratifiedShuffleSplit: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolGroupKFold: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, y: [Int]? = nil, groups: [Int]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + if let groups = groups { + __kwargs["groups"] = RemotePythonValue(groups) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, y: [Int]? = nil, groups: [Int]? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + if let groups = groups { + __kwargs["groups"] = RemotePythonValue(groups) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolTimeSeriesSplit: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRepeatedKFold: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolRepeatedStratifiedKFold: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func split(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "split", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_n_splits(X: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_splits", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolGridSearchCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> PoolGridSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolGridSearchCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolRandomizedSearchCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> PoolRandomizedSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRandomizedSearchCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolHalvingGridSearchCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> PoolHalvingGridSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolHalvingGridSearchCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolHalvingRandomSearchCV: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> PoolHalvingRandomSearchCV { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolHalvingRandomSearchCV(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: any PythonConvertible, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolParameterGrid: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } +} + + +public struct PoolParameterSampler: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } +} + +// MARK: - Sklearn.NaiveBayes + +extension PoolSklearn { + /// Pool-bound `sklearn.naive_bayes` namespace. Calls run on this pool. + public var naivebayes: PoolSklearnNaivebayes { PoolSklearnNaivebayes(pool: pool) } +} + +public struct PoolSklearnNaivebayes: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func GaussianNB(priors: [Double]? = nil, var_smoothing: Double? = nil, worker: Int? = nil) async throws -> PoolGaussianNB { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let priors = priors { + __kwargs["priors"] = RemotePythonValue(priors) + } + if let var_smoothing = var_smoothing { + __kwargs["var_smoothing"] = RemotePythonValue(var_smoothing) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "GaussianNB", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "GaussianNB", args: __args, kwargs: __kwargs) + } + return PoolGaussianNB(pool: pool, owned: __owned) + } + + public func MultinomialNB(alpha: Double? = nil, force_alpha: Bool? = nil, fit_prior: Bool? = nil, class_prior: [Double]? = nil, worker: Int? = nil) async throws -> PoolMultinomialNB { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let force_alpha = force_alpha { + __kwargs["force_alpha"] = RemotePythonValue(force_alpha) + } + if let fit_prior = fit_prior { + __kwargs["fit_prior"] = RemotePythonValue(fit_prior) + } + if let class_prior = class_prior { + __kwargs["class_prior"] = RemotePythonValue(class_prior) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "MultinomialNB", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "MultinomialNB", args: __args, kwargs: __kwargs) + } + return PoolMultinomialNB(pool: pool, owned: __owned) + } + + public func BernoulliNB(alpha: Double? = nil, force_alpha: Bool? = nil, binarize: Double? = nil, fit_prior: Bool? = nil, class_prior: [Double]? = nil, worker: Int? = nil) async throws -> PoolBernoulliNB { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let force_alpha = force_alpha { + __kwargs["force_alpha"] = RemotePythonValue(force_alpha) + } + if let binarize = binarize { + __kwargs["binarize"] = RemotePythonValue(binarize) + } + if let fit_prior = fit_prior { + __kwargs["fit_prior"] = RemotePythonValue(fit_prior) + } + if let class_prior = class_prior { + __kwargs["class_prior"] = RemotePythonValue(class_prior) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "BernoulliNB", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "BernoulliNB", args: __args, kwargs: __kwargs) + } + return PoolBernoulliNB(pool: pool, owned: __owned) + } + + public func ComplementNB(alpha: Double? = nil, force_alpha: Bool? = nil, fit_prior: Bool? = nil, class_prior: [Double]? = nil, norm: Bool? = nil, worker: Int? = nil) async throws -> PoolComplementNB { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let force_alpha = force_alpha { + __kwargs["force_alpha"] = RemotePythonValue(force_alpha) + } + if let fit_prior = fit_prior { + __kwargs["fit_prior"] = RemotePythonValue(fit_prior) + } + if let class_prior = class_prior { + __kwargs["class_prior"] = RemotePythonValue(class_prior) + } + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "ComplementNB", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "ComplementNB", args: __args, kwargs: __kwargs) + } + return PoolComplementNB(pool: pool, owned: __owned) + } + + public func CategoricalNB(alpha: Double? = nil, force_alpha: Bool? = nil, fit_prior: Bool? = nil, class_prior: [Double]? = nil, min_categories: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolCategoricalNB { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let alpha = alpha { + __kwargs["alpha"] = RemotePythonValue(alpha) + } + if let force_alpha = force_alpha { + __kwargs["force_alpha"] = RemotePythonValue(force_alpha) + } + if let fit_prior = fit_prior { + __kwargs["fit_prior"] = RemotePythonValue(fit_prior) + } + if let class_prior = class_prior { + __kwargs["class_prior"] = RemotePythonValue(class_prior) + } + if let min_categories = min_categories { + __kwargs["min_categories"] = RemotePythonValue(min_categories) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "CategoricalNB", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.naive_bayes", function: "CategoricalNB", args: __args, kwargs: __kwargs) + } + return PoolCategoricalNB(pool: pool, owned: __owned) + } +} + + +public struct PoolGaussianNB: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolGaussianNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolGaussianNB(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolGaussianNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolGaussianNB(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolMultinomialNB: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolMultinomialNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMultinomialNB(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolMultinomialNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolMultinomialNB(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolBernoulliNB: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolBernoulliNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBernoulliNB(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolBernoulliNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolBernoulliNB(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolComplementNB: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolComplementNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolComplementNB(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolComplementNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolComplementNB(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolCategoricalNB: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolCategoricalNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolCategoricalNB(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, y: [Int], classes: [Int]? = nil, worker: Int? = nil) async throws -> PoolCategoricalNB { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolCategoricalNB(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + +// MARK: - Sklearn.Neighbors + +extension PoolSklearn { + /// Pool-bound `sklearn.neighbors` namespace. Calls run on this pool. + public var neighbors: PoolSklearnNeighbors { PoolSklearnNeighbors(pool: pool) } +} + +public struct PoolSklearnNeighbors: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func kneighbors_graph(X: any PythonConvertible, n_neighbors: Int, mode: String? = nil, metric: String? = nil, p: Int? = nil, metric_params: (any PythonConvertible)? = nil, include_self: Bool? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(n_neighbors)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let include_self = include_self { + __kwargs["include_self"] = RemotePythonValue(include_self) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "kneighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "kneighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func radius_neighbors_graph(X: any PythonConvertible, radius: Double, mode: String? = nil, metric: String? = nil, p: Int? = nil, metric_params: (any PythonConvertible)? = nil, include_self: Bool? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(radius)) + var __kwargs: [String: RemotePythonValue] = [:] + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let include_self = include_self { + __kwargs["include_self"] = RemotePythonValue(include_self) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "radius_neighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "radius_neighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func KNeighborsClassifier(n_neighbors: Int? = nil, weights: String? = nil, algorithm: String? = nil, leaf_size: Int? = nil, p: Int? = nil, metric: String? = nil, metric_params: (any PythonConvertible)? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolKNeighborsClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "KNeighborsClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "KNeighborsClassifier", args: __args, kwargs: __kwargs) + } + return PoolKNeighborsClassifier(pool: pool, owned: __owned) + } + + public func KNeighborsRegressor(n_neighbors: Int? = nil, weights: String? = nil, algorithm: String? = nil, leaf_size: Int? = nil, p: Int? = nil, metric: String? = nil, metric_params: (any PythonConvertible)? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolKNeighborsRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "KNeighborsRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "KNeighborsRegressor", args: __args, kwargs: __kwargs) + } + return PoolKNeighborsRegressor(pool: pool, owned: __owned) + } + + public func RadiusNeighborsClassifier(radius: Double? = nil, weights: String? = nil, algorithm: String? = nil, leaf_size: Int? = nil, p: Int? = nil, metric: String? = nil, outlier_label: (any PythonConvertible)? = nil, metric_params: (any PythonConvertible)? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolRadiusNeighborsClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let outlier_label = outlier_label { + __kwargs["outlier_label"] = RemotePythonValue(outlier_label) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "RadiusNeighborsClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "RadiusNeighborsClassifier", args: __args, kwargs: __kwargs) + } + return PoolRadiusNeighborsClassifier(pool: pool, owned: __owned) + } + + public func RadiusNeighborsRegressor(radius: Double? = nil, weights: String? = nil, algorithm: String? = nil, leaf_size: Int? = nil, p: Int? = nil, metric: String? = nil, metric_params: (any PythonConvertible)? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolRadiusNeighborsRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let weights = weights { + __kwargs["weights"] = RemotePythonValue(weights) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "RadiusNeighborsRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "RadiusNeighborsRegressor", args: __args, kwargs: __kwargs) + } + return PoolRadiusNeighborsRegressor(pool: pool, owned: __owned) + } + + public func NearestNeighbors(n_neighbors: Int? = nil, radius: Double? = nil, algorithm: String? = nil, leaf_size: Int? = nil, metric: String? = nil, p: Int? = nil, metric_params: (any PythonConvertible)? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolNearestNeighbors { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "NearestNeighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "NearestNeighbors", args: __args, kwargs: __kwargs) + } + return PoolNearestNeighbors(pool: pool, owned: __owned) + } + + public func NearestCentroid(metric: String? = nil, shrink_threshold: Double? = nil, worker: Int? = nil) async throws -> PoolNearestCentroid { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let shrink_threshold = shrink_threshold { + __kwargs["shrink_threshold"] = RemotePythonValue(shrink_threshold) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "NearestCentroid", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "NearestCentroid", args: __args, kwargs: __kwargs) + } + return PoolNearestCentroid(pool: pool, owned: __owned) + } + + public func BallTree(X: any PythonConvertible, leaf_size: Int? = nil, metric: String? = nil, worker: Int? = nil) async throws -> PoolBallTree { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "BallTree", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "BallTree", args: __args, kwargs: __kwargs) + } + return PoolBallTree(pool: pool, owned: __owned) + } + + public func KDTree(X: any PythonConvertible, leaf_size: Int? = nil, metric: String? = nil, worker: Int? = nil) async throws -> PoolKDTree { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "KDTree", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "KDTree", args: __args, kwargs: __kwargs) + } + return PoolKDTree(pool: pool, owned: __owned) + } + + public func LocalOutlierFactor(n_neighbors: Int? = nil, algorithm: String? = nil, leaf_size: Int? = nil, metric: String? = nil, p: Int? = nil, metric_params: (any PythonConvertible)? = nil, contamination: String? = nil, novelty: Bool? = nil, n_jobs: Int? = nil, worker: Int? = nil) async throws -> PoolLocalOutlierFactor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let algorithm = algorithm { + __kwargs["algorithm"] = RemotePythonValue(algorithm) + } + if let leaf_size = leaf_size { + __kwargs["leaf_size"] = RemotePythonValue(leaf_size) + } + if let metric = metric { + __kwargs["metric"] = RemotePythonValue(metric) + } + if let p = p { + __kwargs["p"] = RemotePythonValue(p) + } + if let metric_params = metric_params { + __kwargs["metric_params"] = RemotePythonValue(metric_params) + } + if let contamination = contamination { + __kwargs["contamination"] = RemotePythonValue(contamination) + } + if let novelty = novelty { + __kwargs["novelty"] = RemotePythonValue(novelty) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "LocalOutlierFactor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.neighbors", function: "LocalOutlierFactor", args: __args, kwargs: __kwargs) + } + return PoolLocalOutlierFactor(pool: pool, owned: __owned) + } +} + + +public struct PoolKNeighborsClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolKNeighborsClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolKNeighborsClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func kneighbors(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, return_distance: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kneighbors_graph(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, mode: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolKNeighborsRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolKNeighborsRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolKNeighborsRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func kneighbors(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, return_distance: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kneighbors_graph(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, mode: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolRadiusNeighborsClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolRadiusNeighborsClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRadiusNeighborsClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func radius_neighbors(X: (any PythonConvertible)? = nil, radius: Double? = nil, return_distance: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func radius_neighbors_graph(X: (any PythonConvertible)? = nil, radius: Double? = nil, mode: String? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolRadiusNeighborsRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolRadiusNeighborsRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRadiusNeighborsRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func radius_neighbors(X: (any PythonConvertible)? = nil, radius: Double? = nil, return_distance: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func radius_neighbors_graph(X: (any PythonConvertible)? = nil, radius: Double? = nil, mode: String? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolNearestNeighbors: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolNearestNeighbors { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolNearestNeighbors(pool: pool, owned: __owned) + } + + public func kneighbors(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, return_distance: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kneighbors_graph(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, mode: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func radius_neighbors(X: (any PythonConvertible)? = nil, radius: Double? = nil, return_distance: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func radius_neighbors_graph(X: (any PythonConvertible)? = nil, radius: Double? = nil, mode: String? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let radius = radius { + __kwargs["radius"] = RemotePythonValue(radius) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "radius_neighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolNearestCentroid: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolNearestCentroid { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolNearestCentroid(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolBallTree: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func query(X: any PythonConvertible, k: Int? = nil, return_distance: Bool? = nil, dualtree: Bool? = nil, breadth_first: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let dualtree = dualtree { + __kwargs["dualtree"] = RemotePythonValue(dualtree) + } + if let breadth_first = breadth_first { + __kwargs["breadth_first"] = RemotePythonValue(breadth_first) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "query", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "query", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func query_radius(X: any PythonConvertible, r: Double, return_distance: Bool? = nil, count_only: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(r)) + var __kwargs: [String: RemotePythonValue] = [:] + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let count_only = count_only { + __kwargs["count_only"] = RemotePythonValue(count_only) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "query_radius", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "query_radius", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolKDTree: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func query(X: any PythonConvertible, k: Int? = nil, return_distance: Bool? = nil, dualtree: Bool? = nil, breadth_first: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let k = k { + __kwargs["k"] = RemotePythonValue(k) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let dualtree = dualtree { + __kwargs["dualtree"] = RemotePythonValue(dualtree) + } + if let breadth_first = breadth_first { + __kwargs["breadth_first"] = RemotePythonValue(breadth_first) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "query", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "query", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func query_radius(X: any PythonConvertible, r: Double, return_distance: Bool? = nil, count_only: Bool? = nil, sort_results: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(r)) + var __kwargs: [String: RemotePythonValue] = [:] + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + if let count_only = count_only { + __kwargs["count_only"] = RemotePythonValue(count_only) + } + if let sort_results = sort_results { + __kwargs["sort_results"] = RemotePythonValue(sort_results) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "query_radius", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "query_radius", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolLocalOutlierFactor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolLocalOutlierFactor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLocalOutlierFactor(pool: pool, owned: __owned) + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kneighbors(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, return_distance: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let return_distance = return_distance { + __kwargs["return_distance"] = RemotePythonValue(return_distance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func kneighbors_graph(X: (any PythonConvertible)? = nil, n_neighbors: Int? = nil, mode: String? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let X = X { + __kwargs["X"] = RemotePythonValue(X) + } + if let n_neighbors = n_neighbors { + __kwargs["n_neighbors"] = RemotePythonValue(n_neighbors) + } + if let mode = mode { + __kwargs["mode"] = RemotePythonValue(mode) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "kneighbors_graph", args: __args, kwargs: __kwargs) + } + return __owned + } +} + +// MARK: - Sklearn.Pipeline + +extension PoolSklearn { + /// Pool-bound `sklearn.pipeline` namespace. Calls run on this pool. + public var pipeline: PoolSklearnPipeline { PoolSklearnPipeline(pool: pool) } +} + +public struct PoolSklearnPipeline: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func make_pipeline(_ args: [RemotePythonValue] = [], memory: String? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolPipeline { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let memory = memory { + __kwargs["memory"] = RemotePythonValue(memory) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_pipeline", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_pipeline", args: __args, kwargs: __kwargs) + } + return PoolPipeline(pool: pool, owned: __owned) + } + + public func make_union(_ args: [RemotePythonValue] = [], n_jobs: Int? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolFeatureUnion { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_union", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_union", args: __args, kwargs: __kwargs) + } + return PoolFeatureUnion(pool: pool, owned: __owned) + } + + public func make_column_transformer(_ args: [RemotePythonValue] = [], remainder: String? = nil, sparse_threshold: Double? = nil, n_jobs: Int? = nil, verbose: Bool? = nil, verbose_feature_names_out: Bool? = nil, force_int_remainder_cols: Bool? = nil, worker: Int? = nil) async throws -> PoolColumnTransformer { + var __args: [RemotePythonValue] = [] + __args.append(contentsOf: args) + var __kwargs: [String: RemotePythonValue] = [:] + if let remainder = remainder { + __kwargs["remainder"] = RemotePythonValue(remainder) + } + if let sparse_threshold = sparse_threshold { + __kwargs["sparse_threshold"] = RemotePythonValue(sparse_threshold) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let verbose_feature_names_out = verbose_feature_names_out { + __kwargs["verbose_feature_names_out"] = RemotePythonValue(verbose_feature_names_out) + } + if let force_int_remainder_cols = force_int_remainder_cols { + __kwargs["force_int_remainder_cols"] = RemotePythonValue(force_int_remainder_cols) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_column_transformer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_column_transformer", args: __args, kwargs: __kwargs) + } + return PoolColumnTransformer(pool: pool, owned: __owned) + } + + public func make_column_selector(pattern: String? = nil, dtype_include: (any PythonConvertible)? = nil, dtype_exclude: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let pattern = pattern { + __kwargs["pattern"] = RemotePythonValue(pattern) + } + if let dtype_include = dtype_include { + __kwargs["dtype_include"] = RemotePythonValue(dtype_include) + } + if let dtype_exclude = dtype_exclude { + __kwargs["dtype_exclude"] = RemotePythonValue(dtype_exclude) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_column_selector", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "make_column_selector", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func Pipeline(steps: PyHandle, memory: String? = nil, verbose: Bool? = nil, worker: Int? = nil) async throws -> PoolPipeline { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(steps)) + var __kwargs: [String: RemotePythonValue] = [:] + if let memory = memory { + __kwargs["memory"] = RemotePythonValue(memory) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "Pipeline", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "Pipeline", args: __args, kwargs: __kwargs) + } + return PoolPipeline(pool: pool, owned: __owned) + } + + public func FeatureUnion(transformer_list: PyHandle, n_jobs: Int? = nil, transformer_weights: (any PythonConvertible)? = nil, verbose: Bool? = nil, verbose_feature_names_out: Bool? = nil, worker: Int? = nil) async throws -> PoolFeatureUnion { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(transformer_list)) + var __kwargs: [String: RemotePythonValue] = [:] + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let transformer_weights = transformer_weights { + __kwargs["transformer_weights"] = RemotePythonValue(transformer_weights) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let verbose_feature_names_out = verbose_feature_names_out { + __kwargs["verbose_feature_names_out"] = RemotePythonValue(verbose_feature_names_out) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "FeatureUnion", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "FeatureUnion", args: __args, kwargs: __kwargs) + } + return PoolFeatureUnion(pool: pool, owned: __owned) + } + + public func ColumnTransformer(transformers: PyHandle, remainder: String? = nil, sparse_threshold: Double? = nil, n_jobs: Int? = nil, transformer_weights: (any PythonConvertible)? = nil, verbose: Bool? = nil, verbose_feature_names_out: Bool? = nil, force_int_remainder_cols: Bool? = nil, worker: Int? = nil) async throws -> PoolColumnTransformer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(transformers)) + var __kwargs: [String: RemotePythonValue] = [:] + if let remainder = remainder { + __kwargs["remainder"] = RemotePythonValue(remainder) + } + if let sparse_threshold = sparse_threshold { + __kwargs["sparse_threshold"] = RemotePythonValue(sparse_threshold) + } + if let n_jobs = n_jobs { + __kwargs["n_jobs"] = RemotePythonValue(n_jobs) + } + if let transformer_weights = transformer_weights { + __kwargs["transformer_weights"] = RemotePythonValue(transformer_weights) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let verbose_feature_names_out = verbose_feature_names_out { + __kwargs["verbose_feature_names_out"] = RemotePythonValue(verbose_feature_names_out) + } + if let force_int_remainder_cols = force_int_remainder_cols { + __kwargs["force_int_remainder_cols"] = RemotePythonValue(force_int_remainder_cols) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "ColumnTransformer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "ColumnTransformer", args: __args, kwargs: __kwargs) + } + return PoolColumnTransformer(pool: pool, owned: __owned) + } + + public func TransformedTargetRegressor(regressor: (any PythonConvertible)? = nil, transformer: (any PythonConvertible)? = nil, `func`: (any PythonConvertible)? = nil, inverse_func: (any PythonConvertible)? = nil, check_inverse: Bool? = nil, worker: Int? = nil) async throws -> PoolTransformedTargetRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let regressor = regressor { + __kwargs["regressor"] = RemotePythonValue(regressor) + } + if let transformer = transformer { + __kwargs["transformer"] = RemotePythonValue(transformer) + } + if let `func` = `func` { + __kwargs["func"] = RemotePythonValue(`func`) + } + if let inverse_func = inverse_func { + __kwargs["inverse_func"] = RemotePythonValue(inverse_func) + } + if let check_inverse = check_inverse { + __kwargs["check_inverse"] = RemotePythonValue(check_inverse) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "TransformedTargetRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.pipeline", function: "TransformedTargetRegressor", args: __args, kwargs: __kwargs) + } + return PoolTransformedTargetRegressor(pool: pool, owned: __owned) + } +} + + +public struct PoolPipeline: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolPipeline { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPipeline(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_predict(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_params(deep: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func set_params(extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolPipeline { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs) + } + return PoolPipeline(pool: pool, owned: __owned) + } +} + + +public struct PoolFeatureUnion: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolFeatureUnion { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolFeatureUnion(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_params(deep: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func set_params(extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolFeatureUnion { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs) + } + return PoolFeatureUnion(pool: pool, owned: __owned) + } + + public func get_feature_names_out(input_features: [String]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let input_features = input_features { + __kwargs["input_features"] = RemotePythonValue(input_features) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolColumnTransformer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolColumnTransformer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolColumnTransformer(pool: pool, owned: __owned) + } + + public func fit_transform(X: any PythonConvertible, y: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let y = y { + __kwargs["y"] = RemotePythonValue(y) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_params(deep: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func set_params(extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolColumnTransformer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs) + } + return PoolColumnTransformer(pool: pool, owned: __owned) + } + + public func get_feature_names_out(input_features: [String]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let input_features = input_features { + __kwargs["input_features"] = RemotePythonValue(input_features) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolTransformedTargetRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolTransformedTargetRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolTransformedTargetRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func get_params(deep: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let deep = deep { + __kwargs["deep"] = RemotePythonValue(deep) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_params", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func set_params(extraKwargs: [String: RemotePythonValue] = [:], worker: Int? = nil) async throws -> PoolTransformedTargetRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + __kwargs.merge(extraKwargs, uniquingKeysWith: { _, new in new }) + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "set_params", args: __args, kwargs: __kwargs) + } + return PoolTransformedTargetRegressor(pool: pool, owned: __owned) + } +} + +// MARK: - Sklearn.Preprocessing + +extension PoolSklearn { + /// Pool-bound `sklearn.preprocessing` namespace. Calls run on this pool. + public var preprocessing: PoolSklearnPreprocessing { PoolSklearnPreprocessing(pool: pool) } +} + +public struct PoolSklearnPreprocessing: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func add_dummy_feature(X: any PythonConvertible, value: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let value = value { + __kwargs["value"] = RemotePythonValue(value) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "add_dummy_feature", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "add_dummy_feature", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func binarize(X: any PythonConvertible, threshold: Double? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let threshold = threshold { + __kwargs["threshold"] = RemotePythonValue(threshold) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "binarize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "binarize", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func label_binarize(y: PyHandle, classes: PyHandle, neg_label: Int? = nil, pos_label: Int? = nil, sparse_output: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + __args.append(RemotePythonValue(classes)) + var __kwargs: [String: RemotePythonValue] = [:] + if let neg_label = neg_label { + __kwargs["neg_label"] = RemotePythonValue(neg_label) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let sparse_output = sparse_output { + __kwargs["sparse_output"] = RemotePythonValue(sparse_output) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "label_binarize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "label_binarize", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func maxabs_scale(X: any PythonConvertible, axis: Int? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "maxabs_scale", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "maxabs_scale", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func minmax_scale(X: any PythonConvertible, feature_range: (any PythonConvertible)? = nil, axis: Int? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + if let feature_range = feature_range { + __args.append(RemotePythonValue(feature_range)) + } + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "minmax_scale", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "minmax_scale", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func normalize(X: any PythonConvertible, norm: String? = nil, axis: Int? = nil, copy: Bool? = nil, return_norm: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let return_norm = return_norm { + __kwargs["return_norm"] = RemotePythonValue(return_norm) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "normalize", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "normalize", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func quantile_transform(X: any PythonConvertible, axis: Int? = nil, n_quantiles: Int? = nil, output_distribution: String? = nil, ignore_implicit_zeros: Bool? = nil, subsample: Int? = nil, random_state: Int? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let n_quantiles = n_quantiles { + __kwargs["n_quantiles"] = RemotePythonValue(n_quantiles) + } + if let output_distribution = output_distribution { + __kwargs["output_distribution"] = RemotePythonValue(output_distribution) + } + if let ignore_implicit_zeros = ignore_implicit_zeros { + __kwargs["ignore_implicit_zeros"] = RemotePythonValue(ignore_implicit_zeros) + } + if let subsample = subsample { + __kwargs["subsample"] = RemotePythonValue(subsample) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "quantile_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "quantile_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func power_transform(X: any PythonConvertible, method: String? = nil, standardize: Bool? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let standardize = standardize { + __kwargs["standardize"] = RemotePythonValue(standardize) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "power_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "power_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func robust_scale(X: any PythonConvertible, axis: Int? = nil, with_centering: Bool? = nil, with_scaling: Bool? = nil, quantile_range: (any PythonConvertible)? = nil, copy: Bool? = nil, unit_variance: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let with_centering = with_centering { + __kwargs["with_centering"] = RemotePythonValue(with_centering) + } + if let with_scaling = with_scaling { + __kwargs["with_scaling"] = RemotePythonValue(with_scaling) + } + if let quantile_range = quantile_range { + __kwargs["quantile_range"] = RemotePythonValue(quantile_range) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let unit_variance = unit_variance { + __kwargs["unit_variance"] = RemotePythonValue(unit_variance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "robust_scale", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "robust_scale", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func scale(X: any PythonConvertible, axis: Int? = nil, with_mean: Bool? = nil, with_std: Bool? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + var __kwargs: [String: RemotePythonValue] = [:] + if let axis = axis { + __kwargs["axis"] = RemotePythonValue(axis) + } + if let with_mean = with_mean { + __kwargs["with_mean"] = RemotePythonValue(with_mean) + } + if let with_std = with_std { + __kwargs["with_std"] = RemotePythonValue(with_std) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "scale", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "scale", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func StandardScaler(copy: Bool? = nil, with_mean: Bool? = nil, with_std: Bool? = nil, worker: Int? = nil) async throws -> PoolStandardScaler { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let with_mean = with_mean { + __kwargs["with_mean"] = RemotePythonValue(with_mean) + } + if let with_std = with_std { + __kwargs["with_std"] = RemotePythonValue(with_std) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "StandardScaler", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "StandardScaler", args: __args, kwargs: __kwargs) + } + return PoolStandardScaler(pool: pool, owned: __owned) + } + + public func MinMaxScaler(feature_range: (any PythonConvertible)? = nil, copy: Bool? = nil, clip: Bool? = nil, worker: Int? = nil) async throws -> PoolMinMaxScaler { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let feature_range = feature_range { + __kwargs["feature_range"] = RemotePythonValue(feature_range) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let clip = clip { + __kwargs["clip"] = RemotePythonValue(clip) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "MinMaxScaler", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "MinMaxScaler", args: __args, kwargs: __kwargs) + } + return PoolMinMaxScaler(pool: pool, owned: __owned) + } + + public func MaxAbsScaler(copy: Bool? = nil, worker: Int? = nil) async throws -> PoolMaxAbsScaler { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "MaxAbsScaler", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "MaxAbsScaler", args: __args, kwargs: __kwargs) + } + return PoolMaxAbsScaler(pool: pool, owned: __owned) + } + + public func RobustScaler(with_centering: Bool? = nil, with_scaling: Bool? = nil, quantile_range: (any PythonConvertible)? = nil, copy: Bool? = nil, unit_variance: Bool? = nil, worker: Int? = nil) async throws -> PoolRobustScaler { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let with_centering = with_centering { + __kwargs["with_centering"] = RemotePythonValue(with_centering) + } + if let with_scaling = with_scaling { + __kwargs["with_scaling"] = RemotePythonValue(with_scaling) + } + if let quantile_range = quantile_range { + __kwargs["quantile_range"] = RemotePythonValue(quantile_range) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + if let unit_variance = unit_variance { + __kwargs["unit_variance"] = RemotePythonValue(unit_variance) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "RobustScaler", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "RobustScaler", args: __args, kwargs: __kwargs) + } + return PoolRobustScaler(pool: pool, owned: __owned) + } + + public func Normalizer(norm: String? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolNormalizer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let norm = norm { + __kwargs["norm"] = RemotePythonValue(norm) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "Normalizer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "Normalizer", args: __args, kwargs: __kwargs) + } + return PoolNormalizer(pool: pool, owned: __owned) + } + + public func Binarizer(threshold: Double? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolBinarizer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let threshold = threshold { + __kwargs["threshold"] = RemotePythonValue(threshold) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "Binarizer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "Binarizer", args: __args, kwargs: __kwargs) + } + return PoolBinarizer(pool: pool, owned: __owned) + } + + public func QuantileTransformer(n_quantiles: Int? = nil, output_distribution: String? = nil, ignore_implicit_zeros: Bool? = nil, subsample: Int? = nil, random_state: Int? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolQuantileTransformer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_quantiles = n_quantiles { + __kwargs["n_quantiles"] = RemotePythonValue(n_quantiles) + } + if let output_distribution = output_distribution { + __kwargs["output_distribution"] = RemotePythonValue(output_distribution) + } + if let ignore_implicit_zeros = ignore_implicit_zeros { + __kwargs["ignore_implicit_zeros"] = RemotePythonValue(ignore_implicit_zeros) + } + if let subsample = subsample { + __kwargs["subsample"] = RemotePythonValue(subsample) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "QuantileTransformer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "QuantileTransformer", args: __args, kwargs: __kwargs) + } + return PoolQuantileTransformer(pool: pool, owned: __owned) + } + + public func PowerTransformer(method: String? = nil, standardize: Bool? = nil, copy: Bool? = nil, worker: Int? = nil) async throws -> PoolPowerTransformer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let method = method { + __kwargs["method"] = RemotePythonValue(method) + } + if let standardize = standardize { + __kwargs["standardize"] = RemotePythonValue(standardize) + } + if let copy = copy { + __kwargs["copy"] = RemotePythonValue(copy) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "PowerTransformer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "PowerTransformer", args: __args, kwargs: __kwargs) + } + return PoolPowerTransformer(pool: pool, owned: __owned) + } + + public func SplineTransformer(n_knots: Int? = nil, degree: Int? = nil, knots: String? = nil, extrapolation: String? = nil, include_bias: Bool? = nil, order: String? = nil, sparse_output: Bool? = nil, worker: Int? = nil) async throws -> PoolSplineTransformer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_knots = n_knots { + __kwargs["n_knots"] = RemotePythonValue(n_knots) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let knots = knots { + __kwargs["knots"] = RemotePythonValue(knots) + } + if let extrapolation = extrapolation { + __kwargs["extrapolation"] = RemotePythonValue(extrapolation) + } + if let include_bias = include_bias { + __kwargs["include_bias"] = RemotePythonValue(include_bias) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + if let sparse_output = sparse_output { + __kwargs["sparse_output"] = RemotePythonValue(sparse_output) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "SplineTransformer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "SplineTransformer", args: __args, kwargs: __kwargs) + } + return PoolSplineTransformer(pool: pool, owned: __owned) + } + + public func KBinsDiscretizer(n_bins: Int? = nil, encode: String? = nil, strategy: String? = nil, dtype: (any PythonConvertible)? = nil, subsample: String? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolKBinsDiscretizer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let n_bins = n_bins { + __kwargs["n_bins"] = RemotePythonValue(n_bins) + } + if let encode = encode { + __kwargs["encode"] = RemotePythonValue(encode) + } + if let strategy = strategy { + __kwargs["strategy"] = RemotePythonValue(strategy) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let subsample = subsample { + __kwargs["subsample"] = RemotePythonValue(subsample) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "KBinsDiscretizer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "KBinsDiscretizer", args: __args, kwargs: __kwargs) + } + return PoolKBinsDiscretizer(pool: pool, owned: __owned) + } + + public func LabelEncoder(worker: Int? = nil) async throws -> PoolLabelEncoder { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "LabelEncoder", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "LabelEncoder", args: __args, kwargs: __kwargs) + } + return PoolLabelEncoder(pool: pool, owned: __owned) + } + + public func OrdinalEncoder(categories: String? = nil, dtype: (any PythonConvertible)? = nil, handle_unknown: String? = nil, unknown_value: (any PythonConvertible)? = nil, encoded_missing_value: Double? = nil, min_frequency: Int? = nil, max_categories: Int? = nil, worker: Int? = nil) async throws -> PoolOrdinalEncoder { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let categories = categories { + __kwargs["categories"] = RemotePythonValue(categories) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let handle_unknown = handle_unknown { + __kwargs["handle_unknown"] = RemotePythonValue(handle_unknown) + } + if let unknown_value = unknown_value { + __kwargs["unknown_value"] = RemotePythonValue(unknown_value) + } + if let encoded_missing_value = encoded_missing_value { + __kwargs["encoded_missing_value"] = RemotePythonValue(encoded_missing_value) + } + if let min_frequency = min_frequency { + __kwargs["min_frequency"] = RemotePythonValue(min_frequency) + } + if let max_categories = max_categories { + __kwargs["max_categories"] = RemotePythonValue(max_categories) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "OrdinalEncoder", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "OrdinalEncoder", args: __args, kwargs: __kwargs) + } + return PoolOrdinalEncoder(pool: pool, owned: __owned) + } + + public func OneHotEncoder(categories: String? = nil, drop: String? = nil, sparse_output: Bool? = nil, dtype: (any PythonConvertible)? = nil, handle_unknown: String? = nil, min_frequency: Int? = nil, max_categories: Int? = nil, feature_name_combiner: String? = nil, worker: Int? = nil) async throws -> PoolOneHotEncoder { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let categories = categories { + __kwargs["categories"] = RemotePythonValue(categories) + } + if let drop = drop { + __kwargs["drop"] = RemotePythonValue(drop) + } + if let sparse_output = sparse_output { + __kwargs["sparse_output"] = RemotePythonValue(sparse_output) + } + if let dtype = dtype { + __kwargs["dtype"] = RemotePythonValue(dtype) + } + if let handle_unknown = handle_unknown { + __kwargs["handle_unknown"] = RemotePythonValue(handle_unknown) + } + if let min_frequency = min_frequency { + __kwargs["min_frequency"] = RemotePythonValue(min_frequency) + } + if let max_categories = max_categories { + __kwargs["max_categories"] = RemotePythonValue(max_categories) + } + if let feature_name_combiner = feature_name_combiner { + __kwargs["feature_name_combiner"] = RemotePythonValue(feature_name_combiner) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "OneHotEncoder", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "OneHotEncoder", args: __args, kwargs: __kwargs) + } + return PoolOneHotEncoder(pool: pool, owned: __owned) + } + + public func LabelBinarizer(neg_label: Int? = nil, pos_label: Int? = nil, sparse_output: Bool? = nil, worker: Int? = nil) async throws -> PoolLabelBinarizer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let neg_label = neg_label { + __kwargs["neg_label"] = RemotePythonValue(neg_label) + } + if let pos_label = pos_label { + __kwargs["pos_label"] = RemotePythonValue(pos_label) + } + if let sparse_output = sparse_output { + __kwargs["sparse_output"] = RemotePythonValue(sparse_output) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "LabelBinarizer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "LabelBinarizer", args: __args, kwargs: __kwargs) + } + return PoolLabelBinarizer(pool: pool, owned: __owned) + } + + public func MultiLabelBinarizer(classes: PyHandle? = nil, sparse_output: Bool? = nil, worker: Int? = nil) async throws -> PoolMultiLabelBinarizer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let classes = classes { + __kwargs["classes"] = RemotePythonValue(classes) + } + if let sparse_output = sparse_output { + __kwargs["sparse_output"] = RemotePythonValue(sparse_output) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "MultiLabelBinarizer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "MultiLabelBinarizer", args: __args, kwargs: __kwargs) + } + return PoolMultiLabelBinarizer(pool: pool, owned: __owned) + } + + public func TargetEncoder(categories: String? = nil, target_type: String? = nil, smooth: String? = nil, cv: Int? = nil, shuffle: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolTargetEncoder { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let categories = categories { + __kwargs["categories"] = RemotePythonValue(categories) + } + if let target_type = target_type { + __kwargs["target_type"] = RemotePythonValue(target_type) + } + if let smooth = smooth { + __kwargs["smooth"] = RemotePythonValue(smooth) + } + if let cv = cv { + __kwargs["cv"] = RemotePythonValue(cv) + } + if let shuffle = shuffle { + __kwargs["shuffle"] = RemotePythonValue(shuffle) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "TargetEncoder", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "TargetEncoder", args: __args, kwargs: __kwargs) + } + return PoolTargetEncoder(pool: pool, owned: __owned) + } + + public func PolynomialFeatures(degree: Int? = nil, interaction_only: Bool? = nil, include_bias: Bool? = nil, order: String? = nil, worker: Int? = nil) async throws -> PoolPolynomialFeatures { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let interaction_only = interaction_only { + __kwargs["interaction_only"] = RemotePythonValue(interaction_only) + } + if let include_bias = include_bias { + __kwargs["include_bias"] = RemotePythonValue(include_bias) + } + if let order = order { + __kwargs["order"] = RemotePythonValue(order) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "PolynomialFeatures", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "PolynomialFeatures", args: __args, kwargs: __kwargs) + } + return PoolPolynomialFeatures(pool: pool, owned: __owned) + } + + public func FunctionTransformer(`func`: (any PythonConvertible)? = nil, inverse_func: (any PythonConvertible)? = nil, validate: Bool? = nil, accept_sparse: Bool? = nil, check_inverse: Bool? = nil, feature_names_out: (any PythonConvertible)? = nil, kw_args: (any PythonConvertible)? = nil, inv_kw_args: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolFunctionTransformer { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let `func` = `func` { + __kwargs["func"] = RemotePythonValue(`func`) + } + if let inverse_func = inverse_func { + __kwargs["inverse_func"] = RemotePythonValue(inverse_func) + } + if let validate = validate { + __kwargs["validate"] = RemotePythonValue(validate) + } + if let accept_sparse = accept_sparse { + __kwargs["accept_sparse"] = RemotePythonValue(accept_sparse) + } + if let check_inverse = check_inverse { + __kwargs["check_inverse"] = RemotePythonValue(check_inverse) + } + if let feature_names_out = feature_names_out { + __kwargs["feature_names_out"] = RemotePythonValue(feature_names_out) + } + if let kw_args = kw_args { + __kwargs["kw_args"] = RemotePythonValue(kw_args) + } + if let inv_kw_args = inv_kw_args { + __kwargs["inv_kw_args"] = RemotePythonValue(inv_kw_args) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "FunctionTransformer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "FunctionTransformer", args: __args, kwargs: __kwargs) + } + return PoolFunctionTransformer(pool: pool, owned: __owned) + } + + public func KernelCenterer(worker: Int? = nil) async throws -> PoolKernelCenterer { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "KernelCenterer", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.preprocessing", function: "KernelCenterer", args: __args, kwargs: __kwargs) + } + return PoolKernelCenterer(pool: pool, owned: __owned) + } +} + + +public struct PoolStandardScaler: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolStandardScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolStandardScaler(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolStandardScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolStandardScaler(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolMinMaxScaler: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMinMaxScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMinMaxScaler(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMinMaxScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolMinMaxScaler(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolMaxAbsScaler: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMaxAbsScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMaxAbsScaler(pool: pool, owned: __owned) + } + + public func partial_fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolMaxAbsScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "partial_fit", args: __args, kwargs: __kwargs) + } + return PoolMaxAbsScaler(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolRobustScaler: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolRobustScaler { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolRobustScaler(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolNormalizer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolNormalizer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolNormalizer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolBinarizer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolBinarizer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolBinarizer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolQuantileTransformer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolQuantileTransformer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolQuantileTransformer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolPowerTransformer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolPowerTransformer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPowerTransformer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolSplineTransformer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolSplineTransformer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSplineTransformer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolKBinsDiscretizer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolKBinsDiscretizer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolKBinsDiscretizer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolLabelEncoder: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(y: PyHandle, worker: Int? = nil) async throws -> PoolLabelEncoder { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLabelEncoder(pool: pool, owned: __owned) + } + + public func transform(y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(y: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolOrdinalEncoder: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolOrdinalEncoder { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolOrdinalEncoder(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolOneHotEncoder: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolOneHotEncoder { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolOneHotEncoder(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_feature_names_out(input_features: [String]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let input_features = input_features { + __kwargs["input_features"] = RemotePythonValue(input_features) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolLabelBinarizer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(y: PyHandle, worker: Int? = nil) async throws -> PoolLabelBinarizer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLabelBinarizer(pool: pool, owned: __owned) + } + + public func transform(y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(Y: any PythonConvertible, threshold: Double? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(Y)) + var __kwargs: [String: RemotePythonValue] = [:] + if let threshold = threshold { + __kwargs["threshold"] = RemotePythonValue(threshold) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolMultiLabelBinarizer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(y: PyHandle, worker: Int? = nil) async throws -> PoolMultiLabelBinarizer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolMultiLabelBinarizer(pool: pool, owned: __owned) + } + + public func transform(y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(Y: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(Y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolTargetEncoder: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: PyHandle, worker: Int? = nil) async throws -> PoolTargetEncoder { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolTargetEncoder(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, y: PyHandle, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolPolynomialFeatures: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolPolynomialFeatures { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolPolynomialFeatures(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_feature_names_out(input_features: [String]? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let input_features = input_features { + __kwargs["input_features"] = RemotePythonValue(input_features) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "get_feature_names_out", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolFunctionTransformer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolFunctionTransformer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolFunctionTransformer(pool: pool, owned: __owned) + } + + public func transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func inverse_transform(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "inverse_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolKernelCenterer: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(K: any PythonConvertible, worker: Int? = nil) async throws -> PoolKernelCenterer { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(K)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolKernelCenterer(pool: pool, owned: __owned) + } + + public func transform(K: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(K)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "transform", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_transform(K: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(K)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_transform", args: __args, kwargs: __kwargs) + } + return __owned + } +} + +// MARK: - Sklearn.Svm + +extension PoolSklearn { + /// Pool-bound `sklearn.svm` namespace. Calls run on this pool. + public var svm: PoolSklearnSvm { PoolSklearnSvm(pool: pool) } +} + +public struct PoolSklearnSvm: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func SVC(C: Double? = nil, kernel: String? = nil, degree: Int? = nil, gamma: String? = nil, coef0: Double? = nil, shrinking: Bool? = nil, probability: Bool? = nil, tol: Double? = nil, cache_size: Double? = nil, class_weight: (any PythonConvertible)? = nil, verbose: Bool? = nil, max_iter: Int? = nil, decision_function_shape: String? = nil, break_ties: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolSVC { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let kernel = kernel { + __kwargs["kernel"] = RemotePythonValue(kernel) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let shrinking = shrinking { + __kwargs["shrinking"] = RemotePythonValue(shrinking) + } + if let probability = probability { + __kwargs["probability"] = RemotePythonValue(probability) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let cache_size = cache_size { + __kwargs["cache_size"] = RemotePythonValue(cache_size) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let decision_function_shape = decision_function_shape { + __kwargs["decision_function_shape"] = RemotePythonValue(decision_function_shape) + } + if let break_ties = break_ties { + __kwargs["break_ties"] = RemotePythonValue(break_ties) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "SVC", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "SVC", args: __args, kwargs: __kwargs) + } + return PoolSVC(pool: pool, owned: __owned) + } + + public func SVR(kernel: String? = nil, degree: Int? = nil, gamma: String? = nil, coef0: Double? = nil, tol: Double? = nil, C: Double? = nil, epsilon: Double? = nil, shrinking: Bool? = nil, cache_size: Double? = nil, verbose: Bool? = nil, max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolSVR { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let kernel = kernel { + __kwargs["kernel"] = RemotePythonValue(kernel) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let epsilon = epsilon { + __kwargs["epsilon"] = RemotePythonValue(epsilon) + } + if let shrinking = shrinking { + __kwargs["shrinking"] = RemotePythonValue(shrinking) + } + if let cache_size = cache_size { + __kwargs["cache_size"] = RemotePythonValue(cache_size) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "SVR", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "SVR", args: __args, kwargs: __kwargs) + } + return PoolSVR(pool: pool, owned: __owned) + } + + public func LinearSVC(penalty: String? = nil, loss: String? = nil, dual: String? = nil, tol: Double? = nil, C: Double? = nil, multi_class: String? = nil, fit_intercept: Bool? = nil, intercept_scaling: Double? = nil, class_weight: (any PythonConvertible)? = nil, verbose: Int? = nil, random_state: Int? = nil, max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolLinearSVC { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let penalty = penalty { + __kwargs["penalty"] = RemotePythonValue(penalty) + } + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let dual = dual { + __kwargs["dual"] = RemotePythonValue(dual) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let multi_class = multi_class { + __kwargs["multi_class"] = RemotePythonValue(multi_class) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let intercept_scaling = intercept_scaling { + __kwargs["intercept_scaling"] = RemotePythonValue(intercept_scaling) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "LinearSVC", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "LinearSVC", args: __args, kwargs: __kwargs) + } + return PoolLinearSVC(pool: pool, owned: __owned) + } + + public func LinearSVR(epsilon: Double? = nil, tol: Double? = nil, C: Double? = nil, loss: String? = nil, fit_intercept: Bool? = nil, intercept_scaling: Double? = nil, dual: String? = nil, verbose: Int? = nil, random_state: Int? = nil, max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolLinearSVR { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let epsilon = epsilon { + __kwargs["epsilon"] = RemotePythonValue(epsilon) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let loss = loss { + __kwargs["loss"] = RemotePythonValue(loss) + } + if let fit_intercept = fit_intercept { + __kwargs["fit_intercept"] = RemotePythonValue(fit_intercept) + } + if let intercept_scaling = intercept_scaling { + __kwargs["intercept_scaling"] = RemotePythonValue(intercept_scaling) + } + if let dual = dual { + __kwargs["dual"] = RemotePythonValue(dual) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "LinearSVR", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "LinearSVR", args: __args, kwargs: __kwargs) + } + return PoolLinearSVR(pool: pool, owned: __owned) + } + + public func NuSVC(nu: Double? = nil, kernel: String? = nil, degree: Int? = nil, gamma: String? = nil, coef0: Double? = nil, shrinking: Bool? = nil, probability: Bool? = nil, tol: Double? = nil, cache_size: Double? = nil, class_weight: (any PythonConvertible)? = nil, verbose: Bool? = nil, max_iter: Int? = nil, decision_function_shape: String? = nil, break_ties: Bool? = nil, random_state: Int? = nil, worker: Int? = nil) async throws -> PoolNuSVC { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let nu = nu { + __kwargs["nu"] = RemotePythonValue(nu) + } + if let kernel = kernel { + __kwargs["kernel"] = RemotePythonValue(kernel) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let shrinking = shrinking { + __kwargs["shrinking"] = RemotePythonValue(shrinking) + } + if let probability = probability { + __kwargs["probability"] = RemotePythonValue(probability) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let cache_size = cache_size { + __kwargs["cache_size"] = RemotePythonValue(cache_size) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + if let decision_function_shape = decision_function_shape { + __kwargs["decision_function_shape"] = RemotePythonValue(decision_function_shape) + } + if let break_ties = break_ties { + __kwargs["break_ties"] = RemotePythonValue(break_ties) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "NuSVC", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "NuSVC", args: __args, kwargs: __kwargs) + } + return PoolNuSVC(pool: pool, owned: __owned) + } + + public func NuSVR(nu: Double? = nil, C: Double? = nil, kernel: String? = nil, degree: Int? = nil, gamma: String? = nil, coef0: Double? = nil, shrinking: Bool? = nil, tol: Double? = nil, cache_size: Double? = nil, verbose: Bool? = nil, max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolNuSVR { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let nu = nu { + __kwargs["nu"] = RemotePythonValue(nu) + } + if let C = C { + __kwargs["C"] = RemotePythonValue(C) + } + if let kernel = kernel { + __kwargs["kernel"] = RemotePythonValue(kernel) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let shrinking = shrinking { + __kwargs["shrinking"] = RemotePythonValue(shrinking) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let cache_size = cache_size { + __kwargs["cache_size"] = RemotePythonValue(cache_size) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "NuSVR", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "NuSVR", args: __args, kwargs: __kwargs) + } + return PoolNuSVR(pool: pool, owned: __owned) + } + + public func OneClassSVM(kernel: String? = nil, degree: Int? = nil, gamma: String? = nil, coef0: Double? = nil, tol: Double? = nil, nu: Double? = nil, shrinking: Bool? = nil, cache_size: Double? = nil, verbose: Bool? = nil, max_iter: Int? = nil, worker: Int? = nil) async throws -> PoolOneClassSVM { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let kernel = kernel { + __kwargs["kernel"] = RemotePythonValue(kernel) + } + if let degree = degree { + __kwargs["degree"] = RemotePythonValue(degree) + } + if let gamma = gamma { + __kwargs["gamma"] = RemotePythonValue(gamma) + } + if let coef0 = coef0 { + __kwargs["coef0"] = RemotePythonValue(coef0) + } + if let tol = tol { + __kwargs["tol"] = RemotePythonValue(tol) + } + if let nu = nu { + __kwargs["nu"] = RemotePythonValue(nu) + } + if let shrinking = shrinking { + __kwargs["shrinking"] = RemotePythonValue(shrinking) + } + if let cache_size = cache_size { + __kwargs["cache_size"] = RemotePythonValue(cache_size) + } + if let verbose = verbose { + __kwargs["verbose"] = RemotePythonValue(verbose) + } + if let max_iter = max_iter { + __kwargs["max_iter"] = RemotePythonValue(max_iter) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "OneClassSVM", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.svm", function: "OneClassSVM", args: __args, kwargs: __kwargs) + } + return PoolOneClassSVM(pool: pool, owned: __owned) + } +} + + +public struct PoolSVC: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolSVC { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSVC(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolSVR: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolSVR { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolSVR(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLinearSVC: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolLinearSVC { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLinearSVC(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolLinearSVR: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolLinearSVR { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolLinearSVR(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolNuSVC: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolNuSVC { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolNuSVC(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolNuSVR: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolNuSVR { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolNuSVR(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolOneClassSVM: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, worker: Int? = nil) async throws -> PoolOneClassSVM { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolOneClassSVM(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_function(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_function", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score_samples(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "score_samples", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func fit_predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit_predict", args: __args, kwargs: __kwargs) + } + return __owned + } +} + +// MARK: - Sklearn.Tree + +extension PoolSklearn { + /// Pool-bound `sklearn.tree` namespace. Calls run on this pool. + public var tree: PoolSklearnTree { PoolSklearnTree(pool: pool) } +} + +public struct PoolSklearnTree: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } + + public func export_graphviz(decision_tree: any PythonConvertible, out_file: String? = nil, max_depth: Int? = nil, feature_names: [String]? = nil, class_names: [String]? = nil, label: String? = nil, filled: Bool? = nil, leaves_parallel: Bool? = nil, impurity: Bool? = nil, node_ids: Bool? = nil, proportion: Bool? = nil, rotate: Bool? = nil, rounded: Bool? = nil, special_characters: Bool? = nil, precision: Int? = nil, fontname: String? = nil, worker: Int? = nil) async throws -> String { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(decision_tree)) + var __kwargs: [String: RemotePythonValue] = [:] + if let out_file = out_file { + __kwargs["out_file"] = RemotePythonValue(out_file) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let feature_names = feature_names { + __kwargs["feature_names"] = RemotePythonValue(feature_names) + } + if let class_names = class_names { + __kwargs["class_names"] = RemotePythonValue(class_names) + } + if let label = label { + __kwargs["label"] = RemotePythonValue(label) + } + if let filled = filled { + __kwargs["filled"] = RemotePythonValue(filled) + } + if let leaves_parallel = leaves_parallel { + __kwargs["leaves_parallel"] = RemotePythonValue(leaves_parallel) + } + if let impurity = impurity { + __kwargs["impurity"] = RemotePythonValue(impurity) + } + if let node_ids = node_ids { + __kwargs["node_ids"] = RemotePythonValue(node_ids) + } + if let proportion = proportion { + __kwargs["proportion"] = RemotePythonValue(proportion) + } + if let rotate = rotate { + __kwargs["rotate"] = RemotePythonValue(rotate) + } + if let rounded = rounded { + __kwargs["rounded"] = RemotePythonValue(rounded) + } + if let special_characters = special_characters { + __kwargs["special_characters"] = RemotePythonValue(special_characters) + } + if let precision = precision { + __kwargs["precision"] = RemotePythonValue(precision) + } + if let fontname = fontname { + __kwargs["fontname"] = RemotePythonValue(fontname) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.tree", function: "export_graphviz", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.tree", function: "export_graphviz", args: __args, kwargs: __kwargs) + } + + public func export_text(decision_tree: any PythonConvertible, feature_names: [String]? = nil, class_names: [String]? = nil, max_depth: Int? = nil, spacing: Int? = nil, decimals: Int? = nil, show_weights: Bool? = nil, worker: Int? = nil) async throws -> String { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(decision_tree)) + var __kwargs: [String: RemotePythonValue] = [:] + if let feature_names = feature_names { + __kwargs["feature_names"] = RemotePythonValue(feature_names) + } + if let class_names = class_names { + __kwargs["class_names"] = RemotePythonValue(class_names) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let spacing = spacing { + __kwargs["spacing"] = RemotePythonValue(spacing) + } + if let decimals = decimals { + __kwargs["decimals"] = RemotePythonValue(decimals) + } + if let show_weights = show_weights { + __kwargs["show_weights"] = RemotePythonValue(show_weights) + } + if let worker { + return try await pool.invokeResult(module: "sklearn.tree", function: "export_text", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.invokeResult(module: "sklearn.tree", function: "export_text", args: __args, kwargs: __kwargs) + } + + public func plot_tree(decision_tree: any PythonConvertible, max_depth: Int? = nil, feature_names: [String]? = nil, class_names: [String]? = nil, label: String? = nil, filled: Bool? = nil, impurity: Bool? = nil, node_ids: Bool? = nil, proportion: Bool? = nil, rounded: Bool? = nil, precision: Int? = nil, ax: (any PythonConvertible)? = nil, fontsize: Int? = nil, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(decision_tree)) + var __kwargs: [String: RemotePythonValue] = [:] + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let feature_names = feature_names { + __kwargs["feature_names"] = RemotePythonValue(feature_names) + } + if let class_names = class_names { + __kwargs["class_names"] = RemotePythonValue(class_names) + } + if let label = label { + __kwargs["label"] = RemotePythonValue(label) + } + if let filled = filled { + __kwargs["filled"] = RemotePythonValue(filled) + } + if let impurity = impurity { + __kwargs["impurity"] = RemotePythonValue(impurity) + } + if let node_ids = node_ids { + __kwargs["node_ids"] = RemotePythonValue(node_ids) + } + if let proportion = proportion { + __kwargs["proportion"] = RemotePythonValue(proportion) + } + if let rounded = rounded { + __kwargs["rounded"] = RemotePythonValue(rounded) + } + if let precision = precision { + __kwargs["precision"] = RemotePythonValue(precision) + } + if let ax = ax { + __kwargs["ax"] = RemotePythonValue(ax) + } + if let fontsize = fontsize { + __kwargs["fontsize"] = RemotePythonValue(fontsize) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "plot_tree", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "plot_tree", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func DecisionTreeClassifier(criterion: String? = nil, splitter: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: (any PythonConvertible)? = nil, random_state: Int? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, class_weight: (any PythonConvertible)? = nil, ccp_alpha: Double? = nil, monotonic_cst: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDecisionTreeClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let splitter = splitter { + __kwargs["splitter"] = RemotePythonValue(splitter) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + if let monotonic_cst = monotonic_cst { + __kwargs["monotonic_cst"] = RemotePythonValue(monotonic_cst) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "DecisionTreeClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "DecisionTreeClassifier", args: __args, kwargs: __kwargs) + } + return PoolDecisionTreeClassifier(pool: pool, owned: __owned) + } + + public func DecisionTreeRegressor(criterion: String? = nil, splitter: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: (any PythonConvertible)? = nil, random_state: Int? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, ccp_alpha: Double? = nil, monotonic_cst: (any PythonConvertible)? = nil, worker: Int? = nil) async throws -> PoolDecisionTreeRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let splitter = splitter { + __kwargs["splitter"] = RemotePythonValue(splitter) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + if let monotonic_cst = monotonic_cst { + __kwargs["monotonic_cst"] = RemotePythonValue(monotonic_cst) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "DecisionTreeRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "DecisionTreeRegressor", args: __args, kwargs: __kwargs) + } + return PoolDecisionTreeRegressor(pool: pool, owned: __owned) + } + + public func ExtraTreeClassifier(criterion: String? = nil, splitter: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: String? = nil, random_state: Int? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, class_weight: (any PythonConvertible)? = nil, ccp_alpha: Double? = nil, worker: Int? = nil) async throws -> PoolExtraTreeClassifier { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let splitter = splitter { + __kwargs["splitter"] = RemotePythonValue(splitter) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let class_weight = class_weight { + __kwargs["class_weight"] = RemotePythonValue(class_weight) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "ExtraTreeClassifier", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "ExtraTreeClassifier", args: __args, kwargs: __kwargs) + } + return PoolExtraTreeClassifier(pool: pool, owned: __owned) + } + + public func ExtraTreeRegressor(criterion: String? = nil, splitter: String? = nil, max_depth: Int? = nil, min_samples_split: Int? = nil, min_samples_leaf: Int? = nil, min_weight_fraction_leaf: Double? = nil, max_features: Double? = nil, random_state: Int? = nil, max_leaf_nodes: Int? = nil, min_impurity_decrease: Double? = nil, ccp_alpha: Double? = nil, worker: Int? = nil) async throws -> PoolExtraTreeRegressor { + let __args: [RemotePythonValue] = [] + var __kwargs: [String: RemotePythonValue] = [:] + if let criterion = criterion { + __kwargs["criterion"] = RemotePythonValue(criterion) + } + if let splitter = splitter { + __kwargs["splitter"] = RemotePythonValue(splitter) + } + if let max_depth = max_depth { + __kwargs["max_depth"] = RemotePythonValue(max_depth) + } + if let min_samples_split = min_samples_split { + __kwargs["min_samples_split"] = RemotePythonValue(min_samples_split) + } + if let min_samples_leaf = min_samples_leaf { + __kwargs["min_samples_leaf"] = RemotePythonValue(min_samples_leaf) + } + if let min_weight_fraction_leaf = min_weight_fraction_leaf { + __kwargs["min_weight_fraction_leaf"] = RemotePythonValue(min_weight_fraction_leaf) + } + if let max_features = max_features { + __kwargs["max_features"] = RemotePythonValue(max_features) + } + if let random_state = random_state { + __kwargs["random_state"] = RemotePythonValue(random_state) + } + if let max_leaf_nodes = max_leaf_nodes { + __kwargs["max_leaf_nodes"] = RemotePythonValue(max_leaf_nodes) + } + if let min_impurity_decrease = min_impurity_decrease { + __kwargs["min_impurity_decrease"] = RemotePythonValue(min_impurity_decrease) + } + if let ccp_alpha = ccp_alpha { + __kwargs["ccp_alpha"] = RemotePythonValue(ccp_alpha) + } + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "ExtraTreeRegressor", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.invokeOwned(module: "sklearn.tree", function: "ExtraTreeRegressor", args: __args, kwargs: __kwargs) + } + return PoolExtraTreeRegressor(pool: pool, owned: __owned) + } +} + + +public struct PoolDecisionTreeClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolDecisionTreeClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolDecisionTreeClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_depth(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs) + } + + public func get_n_leaves(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs) + } + + public func cost_complexity_pruning_path(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cost_complexity_pruning_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cost_complexity_pruning_path", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolDecisionTreeRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolDecisionTreeRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolDecisionTreeRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_depth(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs) + } + + public func get_n_leaves(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs) + } + + public func cost_complexity_pruning_path(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "cost_complexity_pruning_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "cost_complexity_pruning_path", args: __args, kwargs: __kwargs) + } + return __owned + } +} + + +public struct PoolExtraTreeClassifier: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> PoolExtraTreeClassifier { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolExtraTreeClassifier(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func predict_log_proba(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict_log_proba", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Int], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_depth(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs) + } + + public func get_n_leaves(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs) + } +} + + +public struct PoolExtraTreeRegressor: Sendable, HandleConvertible { + public let pool: PythonProcessPool + public let owned: OwnedPyHandle + public var handle: PyHandle { owned.handle } + public init(pool: PythonProcessPool, owned: OwnedPyHandle) { + self.pool = pool + self.owned = owned + } + + public func fit(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> PoolExtraTreeRegressor { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "fit", args: __args, kwargs: __kwargs) + } + return PoolExtraTreeRegressor(pool: pool, owned: __owned) + } + + public func predict(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "predict", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func score(X: any PythonConvertible, y: [Double], worker: Int? = nil) async throws -> Double { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + __args.append(RemotePythonValue(y)) + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "score", args: __args, kwargs: __kwargs) + } + + public func apply(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "apply", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func decision_path(X: any PythonConvertible, worker: Int? = nil) async throws -> OwnedPyHandle { + var __args: [RemotePythonValue] = [] + __args.append(RemotePythonValue(X)) + let __kwargs: [String: RemotePythonValue] = [:] + let __owned: OwnedPyHandle + if let worker { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs, worker: worker) + } else { + __owned = try await pool.methodOwned(handle: self, name: "decision_path", args: __args, kwargs: __kwargs) + } + return __owned + } + + public func get_depth(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_depth", args: __args, kwargs: __kwargs) + } + + public func get_n_leaves(worker: Int? = nil) async throws -> Int { + let __args: [RemotePythonValue] = [] + let __kwargs: [String: RemotePythonValue] = [:] + if let worker { + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs, worker: worker) + } + return try await pool.methodResult(handle: self, name: "get_n_leaves", args: __args, kwargs: __kwargs) + } +} + +// MARK: - Sklearn (synthesized root namespace) + +extension PythonProcessPool { + /// Pool-bound `sklearn` namespace. Calls run on this pool. + public var sklearn: PoolSklearn { PoolSklearn(pool: self) } +} + +public struct PoolSklearn: Sendable { + public let pool: PythonProcessPool + public init(pool: PythonProcessPool) { self.pool = pool } +} + diff --git a/Examples/Studio/Sources/studio-cli/main.swift b/Examples/Studio/Sources/studio-cli/main.swift new file mode 100644 index 0000000..dbf8942 --- /dev/null +++ b/Examples/Studio/Sources/studio-cli/main.swift @@ -0,0 +1,156 @@ +// studio-cli — the terminal face of SwiftPython Studio. +// +// Races several scikit-learn classifiers across Python worker processes, driven +// entirely from Swift through the canonical `pool.` surface. This is the +// "behind the curtain" view: same engine as the app, pure terminal output. + +import Foundation +import SwiftPythonRuntime +import SwiftPythonKit + +// MARK: - ANSI + +enum A { + static let reset = "\u{001B}[0m" + static let bold = "\u{001B}[1m" + static let dim = "\u{001B}[2m" + static let orange = "\u{001B}[38;5;208m" + static let blue = "\u{001B}[38;5;75m" + static let green = "\u{001B}[38;5;78m" + static let purple = "\u{001B}[38;5;141m" + static let grey = "\u{001B}[38;5;245m" +} +let modelColor = [A.orange, A.blue, A.purple, A.green] + +func bar(_ value: Double, width: Int = 28) -> String { + let filled = Int((value * Double(width)).rounded()) + return String(repeating: "█", count: max(0, filled)) + String(repeating: "·", count: max(0, width - filled)) +} + +// MARK: - Deterministic dataset (SplitMix64) + +struct SplitMix64 { + var state: UInt64 + mutating func next() -> UInt64 { + state &+= 0x9E3779B97F4A7C15 + var z = state + z = (z ^ (z >> 30)) &* 0xBF58476D1CE4E5B9 + z = (z ^ (z >> 27)) &* 0x94D049BB133111EB + return z ^ (z >> 31) + } + mutating func unit() -> Double { Double(next() >> 11) * (1.0 / 9007199254740992.0) } + mutating func gauss() -> Double { + let u1 = max(unit(), 1e-12), u2 = unit() + return (-2.0 * log(u1)).squareRoot() * cos(2.0 * .pi * u2) + } +} + +func makeDataset(samples: Int, features: Int, classes: Int, seed: UInt64) + -> ([[Double]], [Int], [[Double]], [Int]) { + var rng = SplitMix64(state: seed) + var centroids: [[Double]] = [] + for c in 0.. Double } + +let specs: [Spec] = [ + Spec(name: "Random Forest", worker: 0, color: modelColor[0], + code: "pool.sklearn.ensemble.RandomForestClassifier(n_estimators: 300)") { p, xtr, ytr, xte, yte, w in + let m = try await p.sklearn.ensemble.RandomForestClassifier(n_estimators: 300, random_state: 0, worker: w) + return try await m.fit(X: xtr, y: ytr, worker: w).score(X: xte, y: yte, worker: w) + }, + Spec(name: "Gradient Boosting", worker: 1, color: modelColor[1], + code: "pool.sklearn.ensemble.GradientBoostingClassifier(n_estimators: 200)") { p, xtr, ytr, xte, yte, w in + let m = try await p.sklearn.ensemble.GradientBoostingClassifier(n_estimators: 200, random_state: 0, worker: w) + return try await m.fit(X: xtr, y: ytr, worker: w).score(X: xte, y: yte, worker: w) + }, + Spec(name: "Logistic Regression", worker: 2, color: modelColor[2], + code: "pool.sklearn.linearmodel.LogisticRegression(max_iter: 2000)") { p, xtr, ytr, xte, yte, w in + let m = try await p.sklearn.linearmodel.LogisticRegression(random_state: 0, max_iter: 2000, worker: w) + return try await m.fit(X: xtr, y: ytr, worker: w).score(X: xte, y: yte, worker: w) + }, + Spec(name: "K-Neighbors", worker: 3, color: modelColor[3], + code: "pool.sklearn.neighbors.KNeighborsClassifier(n_neighbors: 7)") { p, xtr, ytr, xte, yte, w in + let m = try await p.sklearn.neighbors.KNeighborsClassifier(n_neighbors: 7, worker: w) + return try await m.fit(X: xtr, y: ytr, worker: w).score(X: xte, y: yte, worker: w) + }, +] + +func workerPath() -> String? { + if let env = ProcessInfo.processInfo.environment["SWIFTPYTHON_WORKER_PATH"], !env.isEmpty { return env } + return nil +} + +// MARK: - Run + +func runStudio() async { + print("") + print(" \(A.bold)\(A.orange)SwiftPython Studio\(A.reset) \(A.grey)· studio-cli\(A.reset)") + print(" \(A.grey)Swift → scikit-learn, across real Python processes.\(A.reset)") + print("") + + let (xtr, ytr, xte, yte) = makeDataset(samples: 1500, features: 12, classes: 3, seed: 0xC0FFEE) + print(" \(A.dim)dataset built in Swift:\(A.reset) 1500×12, 3 classes → train \(ytr.count) / test \(yte.count)") + + do { + StudioRuntime.configureBundledPythonIfPresent() + let wall = Date() + print(" \(A.orange)›\(A.reset) let pool = try await PythonProcessPool(workers: 4)") + let pool = try await PythonProcessPool(workers: 4, workerExecutablePath: workerPath()) + print(" \(A.green)✓\(A.reset) 4 Python workers spawned") + print(" \(A.dim)racing \(specs.count) models in parallel — each pinned to its own worker…\(A.reset)\n") + + let snapTr = xtr, snapYtr = ytr, snapTe = xte, snapYte = yte + var results: [(name: String, color: String, acc: Double, ms: Int)] = [] + + try await withThrowingTaskGroup(of: (Int, Double, Int).self) { group in + for (i, spec) in specs.enumerated() { + group.addTask { + let t0 = Date() + let acc = try await spec.run(pool, snapTr, snapYtr, snapTe, snapYte, spec.worker) + return (i, acc, Int(Date().timeIntervalSince(t0) * 1000)) + } + } + for try await (i, acc, ms) in group { + let s = specs[i] + print(" \(s.color)✓\(A.reset) [w\(s.worker)] \(s.name) → \(A.bold)\(String(format: "%.3f", acc))\(A.reset) \(A.dim)(\(ms) ms)\(A.reset)") + results.append((s.name, s.color, acc, ms)) + } + } + + print("") + for r in results.sorted(by: { $0.acc > $1.acc }) { + let name = r.name.padding(toLength: 20, withPad: " ", startingAt: 0) + print(" \(r.color)\(name)\(A.reset) \(r.color)\(bar(r.acc))\(A.reset) \(String(format: "%.3f", r.acc))") + } + if let best = results.max(by: { $0.acc < $1.acc }) { + print("\n \(A.green)\(A.bold)winner:\(A.reset) \(best.name) \(String(format: "(%.3f)", best.acc))") + } + print(" \(A.dim)total \(Int(Date().timeIntervalSince(wall) * 1000)) ms · handles auto-released\(A.reset)\n") + await pool.shutdown() + fflush(nil) + exit(0) + } catch { + print(" \(A.bold)error:\(A.reset) \(error)") + fflush(nil) + exit(1) + } +} + +await runStudio() diff --git a/Examples/Studio/scripts/build_app.sh b/Examples/Studio/scripts/build_app.sh new file mode 100755 index 0000000..9cd0035 --- /dev/null +++ b/Examples/Studio/scripts/build_app.sh @@ -0,0 +1,140 @@ +#!/usr/bin/env bash +# Assemble SwiftPython Studio.app from the built Swift product. +# +# Modes: +# (default) dev bundle — relies on a system/Homebrew Python 3.13 at runtime. +# --bundle-python copy a relocatable Python + numpy/pandas/scikit-learn/scipy into +# the app so it runs on a machine with no Python (see bundle_python.sh). +# --sign codesign with Developer ID (inside-out: worker + bundled native code, +# then the app), hardened runtime. Required before notarization. +# --open launch the app when done. +# +# Notarization + DMG are separate steps (see SHIPPING.md). +set -euo pipefail + +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +PKG_DIR="$(dirname "$SCRIPT_DIR")" +REPO_DIR="$(cd "$PKG_DIR/../.." && pwd)" # swiftpython-commercial root (has SwiftPythonWorker) +APP_NAME="SwiftPython Studio" +APP_DIR="$PKG_DIR/build/${APP_NAME}.app" +CONFIG="release" +DO_OPEN=0; DO_SIGN=0; DO_BUNDLE_PYTHON=0 +SIGN_ID="${STUDIO_CODESIGN_IDENTITY:-Developer ID Application}" + +for arg in "$@"; do case "$arg" in + --open) DO_OPEN=1 ;; + --sign) DO_SIGN=1 ;; + --bundle-python) DO_BUNDLE_PYTHON=1 ;; + --debug) CONFIG="debug" ;; + *) echo "Unknown argument: $arg" >&2; exit 64 ;; +esac; done + +cd "$PKG_DIR" +echo "==> swift build -c $CONFIG (Studio + studio-cli)" +swift build -c "$CONFIG" --product Studio +swift build -c "$CONFIG" --product studio-cli +BIN_DIR="$(swift build -c "$CONFIG" --show-bin-path)" + +echo "==> assembling $APP_DIR" +rm -rf "$APP_DIR" +mkdir -p "$APP_DIR/Contents/MacOS" "$APP_DIR/Contents/Resources" +cp "$BIN_DIR/Studio" "$APP_DIR/Contents/MacOS/Studio" +cp "$BIN_DIR/studio-cli" "$APP_DIR/Contents/MacOS/studio-cli" + +# App icon +if [ -f "$PKG_DIR/Resources/AppIcon.icns" ]; then + cp "$PKG_DIR/Resources/AppIcon.icns" "$APP_DIR/Contents/Resources/AppIcon.icns" +else + echo "WARNING: Resources/AppIcon.icns not found; app will use the generic icon" >&2 +fi + +# Bundle the worker sidecar so PythonProcessPool auto-discovers it. +if [ -f "$REPO_DIR/SwiftPythonWorker" ]; then + cp "$REPO_DIR/SwiftPythonWorker" "$APP_DIR/Contents/MacOS/SwiftPythonWorker" +else + echo "WARNING: $REPO_DIR/SwiftPythonWorker not found; pool will rely on SWIFTPYTHON_WORKER_PATH" >&2 +fi + +cat > "$APP_DIR/Contents/Info.plist" < + + + + CFBundleExecutableStudio + CFBundleIdentifierdev.swiftpython.studio + CFBundleNameSwiftPython Studio + CFBundleDisplayNameSwiftPython Studio + CFBundleIconFileAppIcon + CFBundleIconNameAppIcon + CFBundlePackageTypeAPPL + CFBundleShortVersionString${STUDIO_APP_SHORT_VERSION:-0.1.0} + CFBundleVersion${STUDIO_APP_BUILD_VERSION:-1} + LSApplicationCategoryTypepublic.app-category.developer-tools + LSMinimumSystemVersion15.0 + NSHighResolutionCapable + NSPrincipalClassNSApplication + + +PLIST + +if [ "$DO_BUNDLE_PYTHON" -eq 1 ]; then + echo "==> bundling Python runtime" + "$SCRIPT_DIR/bundle_python.sh" "$APP_DIR" + + echo "==> relinking binaries to the bundled Python framework (@rpath)" + OLD_PY="/opt/homebrew/opt/python@3.13/Frameworks/Python.framework/Versions/3.13/Python" + NEW_PY="@rpath/Python.framework/Versions/3.13/Python" + for b in Studio studio-cli SwiftPythonWorker; do + bp="$APP_DIR/Contents/MacOS/$b" + [ -f "$bp" ] || continue + install_name_tool -change "$OLD_PY" "$NEW_PY" "$bp" 2>/dev/null || true + install_name_tool -add_rpath "@executable_path/../Frameworks" "$bp" 2>/dev/null || true + done +fi + +if [ "$DO_SIGN" -eq 1 ]; then + echo "==> codesigning (Developer ID, hardened runtime, inside-out)" + ENT="$SCRIPT_DIR/studio.entitlements" + SCAN_DIRS=() + [ -d "$APP_DIR/Contents/Frameworks" ] && SCAN_DIRS+=("$APP_DIR/Contents/Frameworks") + [ -d "$APP_DIR/Contents/Resources/Python" ] && SCAN_DIRS+=("$APP_DIR/Contents/Resources/Python") + + if [ "${#SCAN_DIRS[@]}" -gt 0 ]; then + # 1a) strip executable bits from non-Mach-O Python payloads (Gatekeeper "damaged" guard) + echo " normalizing non-Mach-O executable bits…" + find "${SCAN_DIRS[@]}" -type f -perm -111 2>/dev/null | while IFS= read -r f; do + file "$f" | grep -q 'Mach-O' || chmod -x "$f" + done + # 1b) sign every nested Mach-O (wheel .so, framework dylibs, the Python binary, bin/python3.13) + echo " signing nested Mach-O (this takes a few minutes)…" + count=0 + while IFS= read -r -d '' f; do + if file "$f" | grep -q 'Mach-O'; then + codesign --force --options runtime --timestamp --sign "$SIGN_ID" "$f" + count=$((count + 1)) + fi + done < <(find "${SCAN_DIRS[@]}" -type f \( -name '*.so' -o -name '*.dylib' -o -perm -111 \) -print0 2>/dev/null) + echo " signed $count nested Mach-O files" + fi + + # 2) re-seal the Python framework as a proper versioned bundle (after nested code is signed) + if [ -d "$APP_DIR/Contents/Frameworks/Python.framework" ]; then + codesign --force --options runtime --timestamp --sign "$SIGN_ID" \ + "$APP_DIR/Contents/Frameworks/Python.framework" + fi + + # 3) the worker sidecar + nested CLI (re-sign after install_name_tool relink). + # NOTE: do NOT sign Contents/MacOS/Studio separately — sealing the .app (step 4) + # signs the main executable + applies entitlements. Signing it twice yields an + # "invalid signature" at notarization. + codesign --force --options runtime --timestamp --sign "$SIGN_ID" "$APP_DIR/Contents/MacOS/SwiftPythonWorker" + codesign --force --options runtime --timestamp --sign "$SIGN_ID" "$APP_DIR/Contents/MacOS/studio-cli" + + # 4) the outer app last (no --deep; seals nested code, signs the main executable + entitlements) + codesign --force --options runtime --timestamp \ + --entitlements "$ENT" --sign "$SIGN_ID" "$APP_DIR" + codesign --verify --strict --verbose=2 "$APP_DIR" +fi + +echo "Created: $APP_DIR" +[ "$DO_OPEN" -eq 1 ] && open "$APP_DIR" || true diff --git a/Examples/Studio/scripts/bundle_python.sh b/Examples/Studio/scripts/bundle_python.sh new file mode 100755 index 0000000..45ecdf2 --- /dev/null +++ b/Examples/Studio/scripts/bundle_python.sh @@ -0,0 +1,59 @@ +#!/usr/bin/env bash +# Bundle a self-contained Python 3.13 + the scientific stack into Studio.app so it +# runs on a Mac with NO system Python. Invoked by build_app.sh --bundle-python. +# +# Lean payload (per the launch decision): numpy, pandas, scikit-learn, scipy. +# +# Strategy: copy the Homebrew Python 3.13 framework as a relocatable interpreter, +# then `pip install --target` the wheels into a bundled site-packages. The app +# points the worker at this interpreter via PYTHONHOME/PYTHONPATH (see +# SHIPPING.md "Wiring the worker to the bundled interpreter"). +# +# NOTE: This is the bundling scaffold. The exact PYTHONHOME wiring + the +# inside-out signing of every wheel .so are covered in SHIPPING.md and must be +# verified end-to-end on the release machine (Developer ID + notarytool). +set -euo pipefail + +APP_DIR="${1:?usage: bundle_python.sh }" +PYVER="3.13" +SRC_FRAMEWORK="${STUDIO_PYTHON_FRAMEWORK:-/opt/homebrew/opt/python@${PYVER}/Frameworks/Python.framework}" +WHEELS=("numpy" "pandas" "scikit-learn" "scipy") + +FRAMEWORKS_DIR="$APP_DIR/Contents/Frameworks" +PY_RES_DIR="$APP_DIR/Contents/Resources/Python" +SITE_PKGS="$PY_RES_DIR/site-packages" + +[ -d "$SRC_FRAMEWORK" ] || { echo "Python.framework not found at $SRC_FRAMEWORK" >&2; exit 1; } + +echo "==> copying Python.framework" +mkdir -p "$FRAMEWORKS_DIR" +ditto "$SRC_FRAMEWORK" "$FRAMEWORKS_DIR/Python.framework" + +echo "==> installing wheels into bundled site-packages: ${WHEELS[*]}" +mkdir -p "$SITE_PKGS" +PYBIN="$FRAMEWORKS_DIR/Python.framework/Versions/$PYVER/bin/python$PYVER" +"$PYBIN" -m pip install --upgrade --no-cache-dir --target "$SITE_PKGS" "${WHEELS[@]}" + +echo "==> removing symlinks that escape the framework (Gatekeeper rejects them)" +# e.g. Homebrew's framework links lib/python3.13/site-packages up into the Cellar. +# We use Contents/Resources/Python/site-packages via PYTHONPATH instead. +# realpath resolves even broken/nonexistent targets, so escaping links are caught +# whether or not their destination exists. +FW="$FRAMEWORKS_DIR/Python.framework" +realpath_py() { /usr/bin/python3 -c 'import os,sys;print(os.path.realpath(sys.argv[1]))' "$1" 2>/dev/null; } +FW_REAL="$(realpath_py "$FW")"; [ -z "$FW_REAL" ] && FW_REAL="$FW" +find "$FW" -type l | while IFS= read -r link; do + resolved="$(realpath_py "$link")" + case "$resolved" in + "$FW_REAL"/*|"$FW_REAL") : ;; # resolves inside the framework — keep + *) rm -f "$link"; echo " removed escaping/broken symlink: ${link#$APP_DIR/}" ;; + esac +done + +echo "==> normalizing executable bits on non-Mach-O Python payloads (Gatekeeper)" +find "$FRAMEWORKS_DIR/Python.framework" "$PY_RES_DIR" -type f -perm -111 2>/dev/null | while IFS= read -r f; do + file "$f" | grep -q 'Mach-O' || chmod -x "$f" +done + +echo "==> bundled Python ready under Contents/Frameworks + Contents/Resources/Python" +echo " (build_app.sh --sign will Developer-ID-sign every wheel .so inside-out)" diff --git a/Examples/Studio/scripts/make_icon.py b/Examples/Studio/scripts/make_icon.py new file mode 100644 index 0000000..5923154 --- /dev/null +++ b/Examples/Studio/scripts/make_icon.py @@ -0,0 +1,62 @@ +#!/usr/bin/env python3 +"""Turn square icon artwork into a proper macOS app-icon PNG. + +Center-crops the source to a square, fits it to the macOS 1024 icon grid +(824 body, ~0.2237 corner radius), cuts the rounded-rect with REAL alpha +(transparent corners — no baked checkerboard), and adds a soft drop shadow. + +Usage: make_icon.py [preview.png] +""" +import sys +from PIL import Image, ImageDraw, ImageFilter + +src, out = sys.argv[1], sys.argv[2] +preview = sys.argv[3] if len(sys.argv) > 3 else None + +CANVAS, BODY, RADIUS = 1024, 824, 185 +MARGIN = (CANVAS - BODY) // 2 + +art = Image.open(src).convert("RGBA") +w, h = art.size +# Pad to a square by replicating edge rows/cols (seamless on a gradient bg) so the +# full emblem is preserved — never center-crop, which can clip the artwork. +if w > h: + pad = w - h + top, bot = pad // 2, pad - pad // 2 + sq = Image.new("RGBA", (w, w)) + sq.paste(art.crop((0, 0, w, 1)).resize((w, top)), (0, 0)) + sq.paste(art, (0, top)) + sq.paste(art.crop((0, h - 1, w, h)).resize((w, bot)), (0, top + h)) + art = sq +elif h > w: + pad = h - w + left, right = pad // 2, pad - pad // 2 + sq = Image.new("RGBA", (h, h)) + sq.paste(art.crop((0, 0, 1, h)).resize((left, h)), (0, 0)) + sq.paste(art, (left, 0)) + sq.paste(art.crop((w - 1, 0, w, h)).resize((right, h)), (left + w, 0)) + art = sq +art = art.resize((BODY, BODY), Image.LANCZOS) + +mask = Image.new("L", (BODY, BODY), 0) +ImageDraw.Draw(mask).rounded_rectangle([0, 0, BODY - 1, BODY - 1], radius=RADIUS, fill=255) +art.putalpha(mask) + +canvas = Image.new("RGBA", (CANVAS, CANVAS), (0, 0, 0, 0)) + +shadow = Image.new("RGBA", (CANVAS, CANVAS), (0, 0, 0, 0)) +ImageDraw.Draw(shadow).rounded_rectangle( + [MARGIN, MARGIN + 16, MARGIN + BODY, MARGIN + BODY + 16], radius=RADIUS, fill=(0, 0, 0, 115) +) +shadow = shadow.filter(ImageFilter.GaussianBlur(20)) +canvas = Image.alpha_composite(canvas, shadow) +canvas.alpha_composite(art, (MARGIN, MARGIN)) +canvas.save(out) + +print("size:", canvas.size, + "| top-left alpha:", canvas.getpixel((4, 4))[3], + "| center alpha:", canvas.getpixel((512, 512))[3]) + +if preview: + bg = Image.new("RGBA", (CANVAS, CANVAS), (236, 236, 238, 255)) + Image.alpha_composite(bg, canvas).convert("RGB").save(preview) diff --git a/Examples/Studio/scripts/studio.entitlements b/Examples/Studio/scripts/studio.entitlements new file mode 100644 index 0000000..827c7f4 --- /dev/null +++ b/Examples/Studio/scripts/studio.entitlements @@ -0,0 +1,14 @@ + + + + + + com.apple.security.cs.disable-library-validation + + + com.apple.security.cs.allow-dyld-environment-variables + + +